test: Add SpinValue test

m-laniakea · Nov 19, 2019 · a006552 · a006552
1 parent 97c3e33
commit a006552
Showing 1 changed file with 35 additions and 0 deletions.
diff --git a/tests/test_spin_value.py b/tests/test_spin_value.py
@@ -0,0 +1,35 @@
+import unittest
+
+import stspin
+
+from stspin.constants import (
+    SpinValue
+)
+
+
+class TestUtility(unittest.TestCase):
+
+    def testSpinValue(self) -> None:
+        with self.assertRaises(AssertionError):
+            SpinValue(-1)
+
+        self.assertEqual(SpinValue(1), SpinValue(1))
+        self.assertEqual(SpinValue(8), SpinValue(0) | SpinValue(8))
+        self.assertEqual(SpinValue(0xFF), SpinValue(0xF0) | SpinValue(0x0F))
+
+        self.assertEqual(SpinValue(22) == SpinValue(0o26), True)
+
+        self.assertEqual(f'{SpinValue(22)}', '22')
+
+        self.assertEqual(SpinValue(22), 22)
+        self.assertEqual(SpinValue(16) | 6, 22)
+
+        self.assertEqual(SpinValue(1) == '1', False)
+        self.assertEqual('1' == SpinValue(1), False)
+
+        self.assertEqual(SpinValue(1) == 1, True)
+        self.assertEqual(1 == SpinValue(1), True)
+
+
+if __name__ == '__main__':
+    unittest.main()