|
@@ -0,0 +1,49 @@
|
|
|
+# -*- 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()
|