浏览代码

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 年之前
父节点
当前提交
60a610611f
共有 1 个文件被更改,包括 49 次插入0 次删除
  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()