def test_vector_advection_1d_sub():
"""
Test __sub__
"""
vector_advection_1d_1 = VectorAdvection1D(5)
vector_advection_1d_1.values = np.ones(5)
vector_advection_1d_2 = VectorAdvection1D(5)
vector_advection_1d_2.values = 2 * np.ones(5)
vector_advection_1d_res = vector_advection_1d_2 - vector_advection_1d_1
np.testing.assert_equal(vector_advection_1d_res.values, np.ones(5))
def test_vector_advection_1d_add():
"""
Test __add__
"""
vector_advection_1d_1 = VectorAdvection1D(5)
vector_advection_1d_1.values = np.ones(5)
vector_advection_1d_2 = VectorAdvection1D(5)
vector_advection_1d_2.values = 2 * np.ones(5)
vector_advection_1d_res = vector_advection_1d_1 + vector_advection_1d_2
np.testing.assert_equal(vector_advection_1d_res.values, 3 * np.ones(5))
def test_vector_advection_1d_set_values():
"""
Test the set_values()
"""
vector_advection_1d = VectorAdvection1D(2)
vector_advection_1d.set_values(np.array([1, 2]))
np.testing.assert_equal(vector_advection_1d.values, np.array([1, 2]))
def test_vector_advection_1d_norm():
"""
Test norm()
"""
vector_advection_1d = VectorAdvection1D(5)
vector_advection_1d.values = np.array([1, 2, 3, 4, 5])
np.testing.assert_equal(np.linalg.norm(np.array([1, 2, 3, 4, 5])), vector_advection_1d.norm())
def test_vector_advection_1d_constructor():
"""
Test constructor
"""
vector_advection_1d = VectorAdvection1D(3)
np.testing.assert_equal(vector_advection_1d.values[0], 0)
np.testing.assert_equal(vector_advection_1d.values[1], 0)
np.testing.assert_equal(vector_advection_1d.values[2], 0)
def test_vector_advection_1d_clone_rand():
"""
Test clone_rand()
"""
vector_advection_1d = VectorAdvection1D(2)
vector_advection_1d_clone = vector_advection_1d.clone_rand()
np.testing.assert_equal(True, isinstance(vector_advection_1d_clone, VectorAdvection1D))
np.testing.assert_equal(len(vector_advection_1d_clone.values), 2)
def test_vector_advection_1d_mul():
"""
Test __mul__
"""
vector_advection_1d_1 = VectorAdvection1D(5)
vector_advection_1d_1.values = np.ones(5)
vector_advection_1d_res = vector_advection_1d_1 * 5
np.testing.assert_equal(vector_advection_1d_res.values, np.ones(5)*5)
vector_advection_1d_res = 7 * vector_advection_1d_1
np.testing.assert_equal(vector_advection_1d_res.values, np.ones(5)*7)
vector_advection_1d_1 *= 9
np.testing.assert_equal(vector_advection_1d_1.values, np.ones(5)*9)
def test_vector_advection_1d_get_values():
"""
Test get_values()
"""
vector_advection_1d = VectorAdvection1D(2)
np.testing.assert_equal(vector_advection_1d.get_values(), np.zeros(2))