# -*- 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()