def setUp(self):
self.x = x = N.ones(10, float)
self.y = y = N.ones(10, float) * 2.0
self.z = z = N.linspace(0, 10, 10)
self.s = s = N.ones(10, float)
src = sources.MLineSource()
src.reset(x=x, y=y, z=z, scalars=s)
self.src = src
def test_set_without_scalars_works(self):
# Given
x, y, z, s, src = self.get_data()
src = sources.MLineSource()
src.reset(x=x, y=y, z=z)
# When
src.m_data = VTKDataSource(data=src.dataset)
src.set(y=y + 1)
# Then
self.assertTrue(np.allclose(src.y, y + 1))
# When
src.reset(x=x, y=y + 1, z=z + 1)
# Then
self.assertTrue(np.allclose(src.y, y + 1))
self.assertTrue(np.allclose(src.z, z + 1))