# my_extension2_test.py
import unittest
import my_extension2
class MyExtension2Test(unittest.TestCase):
def test_decimals(self):
import decimal
a = [decimal.Decimal(1), decimal.Decimal(2)]
b = [decimal.Decimal(3), decimal.Decimal(4)]
result = my_extension2.dot_product(a, b)
self.assertEqual(11, result)
def test_not_lists(self):
result1 = my_extension2.dot_product(
(1, 2),
[3, 4],
)
result2 = my_extension2.dot_product(
[1, 2],
(3, 4),
)
result3 = my_extension2.dot_product(
range(1, 3),
range(3, 5),
)
self.assertAlmostEqual(11, result1)
self.assertAlmostEqual(11, result2)
self.assertAlmostEqual(11, result3)
...
if __name__ == "__main__":
unittest.main()