Geografic-Information-System/lib/python/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()