def test_vector_adder(self): add_two = graph.AddTwo() np.testing.assert_array_equal( np.array([5, 7, 9]), add_two.Add( np.array([1, 2, 3]), np.array([4, 5, 6])))
def test_adder(self): add_two = graph.AddTwo() self.assertEqual(3, add_two.Add(1, 2)) self.assertEqual(7, add_two.Add(2, 5))