Geografic-Information-System/scripts/d.polar/testsuite/test_d_polar.py

"""
Created on Fri 20 Nov 2020

@author: Markus Neteler
"""

from grass.gunittest.case import TestCase
from grass.gunittest.main import test
from grass.gunittest.gmodules import SimpleModule

import os


class TestDPolar(TestCase):
    """Test d.polar script"""

    mapName = "aspect"
    epsFile = "polar.eps"

    @classmethod
    def setUpClass(cls):
        """Set region."""
        cls.use_temp_region()
        cls.runModule("g.region", raster=cls.mapName, flags="p")

    @classmethod
    def tearDownClass(cls):
        """Remove temporary region"""
        cls.del_temp_region()

        if os.path.isfile(cls.epsFile):
            os.remove(cls.epsFile)

    def test_d_polar(self):
        """EPS file test"""
        module = SimpleModule("d.polar", map=self.mapName, output=self.epsFile)
        self.assertModule(module)

        self.assertFileExists(filename=self.epsFile)


if __name__ == "__main__":
    test()