(self)
| 85 | scene.remove() |
| 86 | |
| 87 | def test_write_read(self): |
| 88 | smoke = CenteredGrid(1, extrapolation.BOUNDARY, x=32, y=32) |
| 89 | vel = StaggeredGrid(2, 0, x=32, y=32) |
| 90 | # write |
| 91 | scene = Scene.create(DIR) |
| 92 | scene.write(smoke=smoke, vel=vel) |
| 93 | self.assertEqual(1, len(scene.frames)) |
| 94 | self.assertEqual(1, len(scene.complete_frames)) |
| 95 | self.assertEqual(2, len(scene.fieldnames)) |
| 96 | # read single |
| 97 | smoke_ = scene.read('smoke') |
| 98 | vel_ = scene.read('vel') |
| 99 | field.assert_close(smoke, smoke_) |
| 100 | field.assert_close(vel, vel_) |
| 101 | self.assertEqual(smoke.extrapolation, smoke_.extrapolation) |
| 102 | self.assertEqual(vel.extrapolation, vel_.extrapolation) |
| 103 | # read multiple |
| 104 | smoke__, vel__ = scene.read(['smoke', 'vel']) # deprecated |
| 105 | field.assert_close(smoke, smoke__) |
| 106 | field.assert_close(vel, vel__) |
| 107 | smoke__, vel__ = scene.read('smoke', 'vel') |
| 108 | field.assert_close(smoke, smoke__) |
| 109 | field.assert_close(vel, vel__) |
| 110 | # read without Scene |
| 111 | smoke_ = phi.field.read(join(scene.path, "smoke_000000.npz")) |
| 112 | field.assert_close(smoke, smoke_) |
| 113 | scene.remove() |
| 114 | |
| 115 | def test_write_read_batch_matching(self): |
| 116 | smoke = CenteredGrid(1, extrapolation.BOUNDARY, x=32, y=32) * math.random_uniform(batch(count=2)) |
nothing calls this directly
no test coverage detected