def test_write_batch(self):
for scene in Scene.list('data'):
scene.remove()
scene_batch = Scene.create('data', count=2)
self.assertEqual(scene_batch.batch_size, 2)
self.assertEqual(len(Scene.list('data')), 2)
batched_data = np.zeros([2, 4, 4, 1])
unbatched_data = np.ones([1, 4, 4, 1])
scene_batch.write(batched_data, frame=0)
scene_batch.write(unbatched_data, frame=1)
self.assertTrue(os.path.exists('data/sim_000000/unnamed_000000.npz'))
self.assertTrue(os.path.exists('data/sim_000000/unnamed_000001.npz'))
self.assertTrue(os.path.exists('data/sim_000001/unnamed_000000.npz'))
self.assertTrue(os.path.exists('data/sim_000001/unnamed_000001.npz'))
def build_test_database(path='data'):
for scene in Scene.list(path):
scene.remove()
val = 1.0
for _scene_index in range(2):
scene = Scene.create(path)
for t in range(4):
scene.write_sim_frame([np.zeros([1, 4, 4, 1]) + val, np.zeros([1, 5, 5, 2])], ['Density', 'Velocity'], t)
val += 1
def test_read_write_struct(self):
for scene in Scene.list('data'):
scene.remove()
state = Fluid(Domain([4, 4]))
scene = Scene.create('data')
scene.write(state, frame=0)
self.assertTrue(isfile(scene.subpath('density_000000.npz')))
self.assertTrue(isfile(scene.subpath('velocity_000000.npz')))
loaded_state = scene.read(state, frame=0)
self.assertIsInstance(loaded_state, Fluid)
self.assertIsInstance(loaded_state.velocity, StaggeredGrid)
self.assertIsInstance(loaded_state.density, CenteredGrid)
_differences = struct.compare([loaded_state.density, state.density])
self.assertEqual(loaded_state.density, state.density)
print_differences(loaded_state.velocity.data, state.velocity.data)
np.testing.assert_equal(loaded_state.velocity.data[0].data,
state.velocity.data[0].data)
scene.write(np.ones([1, 4, 4, 1]) * 2, frame=1)
self.assertTrue(isfile(scene.subpath('unnamed_000001.npz')))
self.assertEqual(scene.read(None, frame=1)[0, 0, 0, 0], 2)
scene.write([np.ones([1, 4, 4, 1])], ['Ones'], frame=2)
self.assertTrue(isfile(scene.subpath('Ones_000002.npz')))
mystruct = [{
'Two': np.ones([1, 4, 4, 1]) * 2,
'Three': np.ones([1, 4, 4, 1]) * 3
}]
scene.write(mystruct, frame=3)
self.assertTrue(isfile(scene.subpath('0_Three_000003.npz')))
self.assertTrue(isfile(scene.subpath('0_Two_000003.npz')))
loaded_struct = scene.read(mystruct, frame=3)
self.assertIsInstance(loaded_struct, list)
np.testing.assert_equal(mystruct[0]['Two'][0, 0, 0, 0],
loaded_struct[0]['Two'][0, 0, 0, 0])
scene.remove()