square grains exercise

python/grains/grains_test.py

@@ -0,0 +1,63 @@
+import unittest
+
+from grains import square, total
+
+
+# Tests adapted from `problem-specifications//canonical-data.json` @ v1.2.0
+
+class GrainsTest(unittest.TestCase):
+    def test_square_1(self):
+        self.assertEqual(square(1), 1)
+
+    def test_square_2(self):
+        self.assertEqual(square(2), 2)
+
+    def test_square_3(self):
+        self.assertEqual(square(3), 4)
+
+    def test_square_4(self):
+        self.assertEqual(square(4), 8)
+
+    def test_square_16(self):
+        self.assertEqual(square(16), 32768)
+
+    def test_square_32(self):
+        self.assertEqual(square(32), 2147483648)
+
+    def test_square_64(self):
+        self.assertEqual(square(64), 9223372036854775808)
+
+    def test_square_0_raises_exception(self):
+        with self.assertRaisesWithMessage(ValueError):
+            square(0)
+        with self.assertRaisesWithMessage(ValueError):
+            total(0)
+
+    def test_square_negative_raises_exception(self):
+        with self.assertRaisesWithMessage(ValueError):
+            square(-1)
+        with self.assertRaisesWithMessage(ValueError):
+            total(-1)
+
+    def test_square_gt_64_raises_exception(self):
+        with self.assertRaisesWithMessage(ValueError):
+            square(65)
+        with self.assertRaisesWithMessage(ValueError):
+            total(65)
+
+    def test_total(self):
+        self.assertEqual(total(64), 18446744073709551615)
+
+    # Utility functions
+    def setUp(self):
+        try:
+            self.assertRaisesRegex
+        except AttributeError:
+            self.assertRaisesRegex = self.assertRaisesRegexp
+
+    def assertRaisesWithMessage(self, exception):
+        return self.assertRaisesRegex(exception, r".+")
+
+
+if __name__ == '__main__':
+    unittest.main()