Browse Source

pygrass: add test for Flag object

git-svn-id: https://svn.osgeo.org/grass/grass/trunk@65597 15284696-431f-4ddb-bdfa-cd5b030d7da7
Pietro Zambelli 9 năm trước cách đây
mục cha
commit
60a610611f
1 tập tin đã thay đổi với 49 bổ sung0 xóa
  1. 49 0
      lib/python/pygrass/modules/interface/testsuite/test_flag.py

+ 49 - 0
lib/python/pygrass/modules/interface/testsuite/test_flag.py

@@ -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()