mirror of
https://github.com/Xevion/exercism.git
synced 2025-12-06 01:14:56 -06:00
64 lines
1.7 KiB
Python
64 lines
1.7 KiB
Python
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()
|