# my_extension_test.py
import unittest
import my_extension
class MyExtensionTest(unittest.TestCase):
def test_empty(self):
result = my_extension.dot_product([], [])
self.assertAlmostEqual(0, result)
def test_positive_result(self):
result = my_extension.dot_product(
[3, 4, 5],
[-1, 9, -2.5],
)
self.assertAlmostEqual(20.5, result)
...
if __name__ == "__main__":
unittest.main()