def test_1d_world_link(): x, y = r(10), r(10) d1 = Data(label='d1', x=x) d2 = Data(label='d2', y=y) dc = DataCollection([d1, d2]) dc.add_link(LinkSame(d2.get_world_component_id(0), d1.id['x'])) assert d2.get_world_component_id(0) in d1.components np.testing.assert_array_equal(d1[d2.get_world_component_id(0)], x) np.testing.assert_array_equal(d1[d2.get_pixel_component_id(0)], x)
def test_3d_world_link(): """Should be able to grab pixel coords after linking world""" x, y, z = r(10), r(10), r(10) cat = Data(label='cat', x=x, y=y, z=z) im = Data(label='im', inten=r((3, 3, 3))) dc = DataCollection([cat, im]) dc.add_link(LinkSame(im.get_world_component_id(2), cat.id['x'])) dc.add_link(LinkSame(im.get_world_component_id(1), cat.id['y'])) dc.add_link(LinkSame(im.get_world_component_id(0), cat.id['z'])) np.testing.assert_array_equal(cat[im.get_pixel_component_id(2)], x) np.testing.assert_array_equal(cat[im.get_pixel_component_id(1)], y) np.testing.assert_array_equal(cat[im.get_pixel_component_id(0)], z)