def test_add_sub_3vecs(): vec1 = np.array([1, 2.3, 3]) vec2 = np.array([2, 3, 4.3]) assert_array_almost_equal(vec1 - vec2, pf.sub_3vecs(vec1, vec2)) assert_array_almost_equal(vec1 + vec2, pf.add_3vecs(vec1, vec2)) vec2 = [2, -3, 4.3] assert_array_almost_equal(vec1 - vec2, pf.sub_3vecs(vec1, vec2)) assert_array_almost_equal(vec1 + vec2, pf.add_3vecs(vec1, vec2))
def test_add_sub_3vecs(): vec1 = np.array([1, 2.3, 3]) vec2 = np.array([2, 3, 4.3]) assert_array_almost_equal( vec1 - vec2, pf.sub_3vecs(vec1, vec2)) assert_array_almost_equal( vec1 + vec2, pf.add_3vecs(vec1, vec2)) vec2 = [2, -3, 4.3] assert_array_almost_equal( vec1 - vec2, pf.sub_3vecs(vec1, vec2)) assert_array_almost_equal( vec1 + vec2, pf.add_3vecs(vec1, vec2))