Added tests for bitwise operations and some more arithmetics.

Author: krischer

future/tests/test_int.py

@@ -655,6 +655,13 @@ def test_upcasting_to_floats(self):
         self.assertTrue(isinstance(a % 0.5, float))
         self.assertTrue(isinstance(0.5 % a, float))
 
+        # Power with floats.
+        self.assertEqual(1.0 ** a, 1.0)
+        self.assertTrue(isinstance(1.0 ** a, float))
+
+        self.assertEqual(a ** 1.0, a)
+        self.assertTrue(isinstance(a ** 1.0, float))
+
     def test_upcasting_to_complex(self):
         """
         Integers should automatically be upcasted to complex numbers for
@@ -689,6 +696,67 @@ def test_upcasting_to_complex(self):
         self.assertTrue(isinstance(a / 0.5j, complex))
         self.assertTrue(isinstance(0.5j / a, complex))
 
+        # Power with floats.
+        self.assertEqual(5.0j ** int(1), 5.0j)
+        self.assertTrue(isinstance(5.0j ** int(1), complex))
+
+        self.assertEqual(a ** 1.0j, 3.0 ** 1.0j)
+        self.assertTrue(isinstance(a ** 1.0j, complex))
+
+    def test_more_arithmetics(self):
+        """
+        More arithmetic tests to improve test coverage.
+        """
+        a = int(3)
+        b = int(5)
+
+        self.assertEqual(b - a, 2)
+        self.assertTrue(isinstance(b - a, int))
+
+        self.assertEqual(a * b, 15)
+        self.assertTrue(isinstance(a * b, int))
+
+        self.assertEqual(b % a, 2)
+        self.assertTrue(isinstance(b % a, int))
+
+        self.assertEqual(a ** b, 243)
+        self.assertTrue(isinstance(a ** b, int))
+
+    def test_bitwise_operations(self):
+        """
+        Tests bitwise operations.
+        """
+        a = int(3)
+        b = int(1)
+
+        self.assertEqual(a >> b, 1)
+        self.assertEqual(a >> 1, 1)
+        self.assertTrue(isinstance(a >> b, int))
+        self.assertTrue(isinstance(a >> 1, int))
+
+        self.assertEqual(a << b, 6)
+        self.assertEqual(a << 1, 6)
+        self.assertTrue(isinstance(a << b, int))
+        self.assertTrue(isinstance(a << 1, int))
+
+        self.assertEqual(a & b, 1)
+        self.assertEqual(a & 1, 1)
+        self.assertTrue(isinstance(a & b, int))
+        self.assertTrue(isinstance(a & 1, int))
+
+        self.assertEqual(a | b, 3)
+        self.assertEqual(a | 1, 3)
+        self.assertTrue(isinstance(a | b, int))
+        self.assertTrue(isinstance(a | 1, int))
+
+        self.assertEqual(a ^ b, 2)
+        self.assertEqual(a ^ 1, 2)
+        self.assertTrue(isinstance(a ^ b, int))
+        self.assertTrue(isinstance(a ^ 1, int))
+
+        self.assertEqual(~a, -4)
+        self.assertTrue(isinstance(~a, int))
+
 
 if __name__ == "__main__":
     unittest.main()