Geografic-Information-System/scripts/r.mapcalc.simple/testsuite/test_rmapcalcsimple.py

"""
Name:       r.mapcalc.simple test
Purpose:    Tests r.mapcalc.simple and its flags/options.
    
Author:     Markus Neteler
Copyright:  (C) 2018 by Markus Neteler and the GRASS Development Team
Licence:    This program is free software under the GNU General Public
            License (>=v2). Read the file COPYING that comes with GRASS
            for details.
"""
from grass.gunittest.case import TestCase


class TestReport(TestCase):
    @classmethod
    def setUpClass(cls):
        """Use temporary region settings"""
        map_input = "elevation"
        cls.runModule("g.region", raster=map_input)
        cls.use_temp_region()

    @classmethod
    def tearDownClass(cls):
        map_output1 = "test1"
        map_output2 = "test2"
        cls.runModule("g.remove", flags="f", type="raster", name=map_output1)
        cls.runModule("g.remove", flags="f", type="raster", name=map_output2)
        cls.del_temp_region()

    def test_rmapcalcsimple(self):
        """Testing r.mapcalc.simple"""
        map_input = "elevation"
        map_output1 = "test1"
        map_output2 = "test2"

        # test 1
        self.assertModule("r.mapcalc.simple", expression="0", output=map_output1)
        self.assertRasterMinMax(
            map=map_output1, refmin=0, refmax=0, msg="Result must be 0 for all pixels"
        )

        # test 2
        formula = "if(%s > 2000, 1, 0)" % map_input  # expected to be 0
        self.assertModule("r.mapcalc.simple", expression=formula, output=map_output2)
        self.assertRasterMinMax(
            map=map_output2, refmin=0, refmax=0, msg="Result must be 0 for all pixels"
        )


if __name__ == "__main__":
    from grass.gunittest.main import test

    test()