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))