MCPcopy
hub / github.com/tum-pbs/PhiFlow / test_batched_properties

Method test_batched_properties

tests/commit/field/test__scene.py:39–61  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

37 scene.remove()
38
39 def test_batched_properties(self):
40 scenes = Scene.create(DIR, batch(scenes=2))
41 batched = wrap([0, 1], batch(scenes=2))
42 scenes.put_properties(batched=batched,
43 non_batched=-1.,
44 batched_tensor=batched * vec(x=2, y=3),
45 non_batched_tensor=vec(x=2, y=3))
46 s0, s1 = scenes.scenes
47 self.assertIsNone(s0._properties)
48 self.assertEqual(0, s0.properties['batched'])
49 self.assertEqual(1, s1.properties['batched'])
50 self.assertEqual(-1, s0.properties['non_batched'])
51 self.assertEqual(-1, s1.properties['non_batched'])
52 math.assert_close((0, 0), s0.properties['batched_tensor'])
53 math.assert_close((2, 3), s1.properties['batched_tensor'])
54 math.assert_close((2, 3), s0.properties['non_batched_tensor'])
55 math.assert_close((2, 3), s1.properties['non_batched_tensor'])
56 scenes = stack([s0, s1], scenes.shape)
57 math.assert_close(batched, scenes.properties['batched'])
58 math.assert_close(-1, scenes.properties['non_batched'])
59 math.assert_close(batched * vec(x=2, y=3), scenes.properties['batched_tensor'])
60 math.assert_close(vec(x=2, y=3), scenes.properties['non_batched_tensor'])
61 scenes.remove()
62
63 def test_create_remove_at_equality_batch(self):
64 scene = Scene.create(DIR, batch=2, config=3)

Callers

nothing calls this directly

Calls 4

stackFunction · 0.90
createMethod · 0.80
put_propertiesMethod · 0.80
removeMethod · 0.80

Tested by

no test coverage detected