"""
Created on Sun Jun 08 13:20:31 2018

@author: Sanjeet Bhatti
"""

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


class TestRRGB(TestCase):
    """Test r.rgb script"""

    mapName = 'elevation'
    red = 'elevation.r'
    green = 'elevation.g'
    blue = 'elevation.b'

    @classmethod
    def setUpClass(cls):
        """Create maps in a small region."""
        cls.use_temp_region()
        cls.runModule('g.region', raster=cls.mapName, flags='p')

    @classmethod
    def tearDownClass(cls):
        """Remove temporary region"""
        cls.runModule('g.remove', flags='f', type='raster',
                      name=(cls.red, cls.green, cls.blue))
        cls.del_temp_region()

    def test_rgb_maps(self):
        """Generates rgb maps from a raster map test"""
        module = SimpleModule('r.rgb', input=self.mapName, red=self.red,
                              green=self.green, blue=self.blue)
        self.assertModule(module)

        self.assertRasterExists(self.red)
        self.assertRasterExists(self.green)
        self.assertRasterExists(self.blue)

if __name__ == '__main__':
    test()