def test_fake_planet(self):
module = FakePlanetModule(position=(0.5, 90.),
magnitude=6.,
psf_scaling=1.,
interpolation='spline',
name_in='fake',
image_in_tag='read',
psf_in_tag='read',
image_out_tag='fake')
self.pipeline.add_module(module)
self.pipeline.run_module('fake')
data = self.pipeline.get_data('fake')
assert np.allclose(data[0, 50, 50], 0.09860622347589054, rtol=limit, atol=0.)
assert np.allclose(np.mean(data), 9.867026482551375e-05, rtol=limit, atol=0.)
assert data.shape == (40, 101, 101)
def test_fake_planet(self) -> None:
module = FakePlanetModule(position=(0.2, 180.),
magnitude=2.5,
psf_scaling=1.,
interpolation='spline',
name_in='fake',
image_in_tag='adi',
psf_in_tag='psf',
image_out_tag='fake')
self.pipeline.add_module(module)
self.pipeline.run_module('fake')
data = self.pipeline.get_data('fake')
assert np.sum(data) == pytest.approx(21.51956021269913,
rel=self.limit,
abs=0.)
assert data.shape == (10, 21, 21)