Geografic-Information-System/python/grass/pygrass/modules/interface/testsuite/test_flag.py

# -*- coding: utf-8 -*-
"""
Created on Tue Jun 24 09:43:53 2014

@author: pietro
"""
from __future__ import unicode_literals
from grass.gunittest.case import TestCase
from grass.gunittest.main import test

from grass.pygrass.modules.interface.flag import Flag


class TestFlag(TestCase):
    def test_get_bash(self):
        """Test get_bash method"""
        flag = Flag(diz=dict(name="a"))
        self.assertFalse(flag.value)
        self.assertEqual("", flag.get_bash())
        flag.special = True
        self.assertEqual("", flag.get_bash())
        flag.value = True
        self.assertEqual("--a", flag.get_bash())
        flag.special = False
        self.assertEqual("-a", flag.get_bash())

    def test_get_python(self):
        """Test get_python method"""
        flag = Flag(diz=dict(name="a"))
        self.assertFalse(flag.value)
        self.assertEqual("", flag.get_python())
        flag.special = True
        self.assertEqual("", flag.get_python())
        flag.value = True
        self.assertEqual("a=True", flag.get_python())
        flag.special = False
        self.assertEqual("a", flag.get_python())

    def test_bool(self):
        """Test magic __bool__ method"""
        flag = Flag(diz=dict(name="a"))
        flag.value = True
        self.assertTrue(True if flag else False)
        flag.value = False
        self.assertFalse(True if flag else False)


if __name__ == "__main__":
    test()