def test_predict_wprojection_graph(self):
flux_model_graph = delayed(self.get_LSM)(self.vis_graph_list[self.nvis // 2], flux=100.0)
zero_vis_graph_list = create_zero_vis_graph_list(self.vis_graph_list)
predicted_vis_graph_list = create_predict_graph(zero_vis_graph_list, flux_model_graph, wstep=4.0,
kernel='wprojection')
residual_vis_graph_list = create_subtract_vis_graph_list(self.vis_graph_list, predicted_vis_graph_list)
if self.compute:
qa = qa_visibility(self.vis_graph_list[0].compute())
numpy.testing.assert_almost_equal(qa.data['maxabs'], 1600.0, 0)
qa = qa_visibility(predicted_vis_graph_list[0].compute())
numpy.testing.assert_almost_equal(qa.data['maxabs'], 111.8, 0)
qa = qa_visibility(residual_vis_graph_list[0].compute())
numpy.testing.assert_almost_equal(qa.data['maxabs'], 1644.3, 0)
def test_predict_graph(self):
flux_model_graph = delayed(self.get_LSM)(self.vis_graph_list[self.nvis // 2], flux=100.0)
zero_vis_graph_list = create_zero_vis_graph_list(self.vis_graph_list)
predicted_vis_graph_list = create_predict_graph(zero_vis_graph_list, flux_model_graph,
vis_slices=self.vis_slices)
residual_vis_graph_list = create_subtract_vis_graph_list(self.vis_graph_list,
predicted_vis_graph_list)
if self.compute:
qa = qa_visibility(self.vis_graph_list[0].compute())
numpy.testing.assert_almost_equal(qa.data['maxabs'], 1600.0, 0)
qa = qa_visibility(predicted_vis_graph_list[0].compute())
numpy.testing.assert_almost_equal(qa.data['maxabs'], 100.064844507, 0)
qa = qa_visibility(residual_vis_graph_list[0].compute())
numpy.testing.assert_almost_equal(qa.data['maxabs'], 1682.1, 0)