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)