def test_add_elementwise(self): c = Polynomial([0]) a = [Element(0), Element(2), Element(1)] b = [Element(1), Element(2), Element(2)] elements = c._add_elementwise(a, b) self.assertEqual([1, 1, 0], [e.element for e in elements])