finish

Author: Jammy2211

autoarray/inversion/inversion/imaging/mapping.py

@@ -131,7 +131,9 @@ def _mapped_reconstructed_data_dict_from(
                 mask=self.mask,
             )
 
-            mapped_reconstructed_data_dict[linear_obj] = mapped_reconstructed_operated_data
+            mapped_reconstructed_data_dict[linear_obj] = (
+                mapped_reconstructed_operated_data
+            )
 
         return mapped_reconstructed_data_dict
 

test_autoarray/inversion/inversion/test_abstract.py

@@ -394,7 +394,9 @@ def test__mapped_reconstructed_operated_data():
         reconstruction_dict=[None],
     )
 
-    assert (inversion.mapped_reconstructed_operated_data_dict[linear_obj_0] == np.ones(3)).all()
+    assert (
+        inversion.mapped_reconstructed_operated_data_dict[linear_obj_0] == np.ones(3)
+    ).all()
     assert (inversion.mapped_reconstructed_operated_data == np.ones(3)).all()
 
     linear_obj_1 = aa.m.MockLinearObj()
@@ -411,9 +413,12 @@ def test__mapped_reconstructed_operated_data():
         reconstruction_dict=[None, None],
     )
 
-    assert (inversion.mapped_reconstructed_operated_data_dict[linear_obj_0] == np.ones(2)).all()
     assert (
-        inversion.mapped_reconstructed_operated_data_dict[linear_obj_1] == 2.0 * np.ones(2)
+        inversion.mapped_reconstructed_operated_data_dict[linear_obj_0] == np.ones(2)
+    ).all()
+    assert (
+        inversion.mapped_reconstructed_operated_data_dict[linear_obj_1]
+        == 2.0 * np.ones(2)
     ).all()
     assert (inversion.mapped_reconstructed_operated_data == 3.0 * np.ones(2)).all()
 

test_autoarray/inversion/inversion/test_factory.py

@@ -22,7 +22,9 @@ def test__inversion_imaging__via_linear_obj_func_list(masked_imaging_7x7_no_blur
 
     assert isinstance(inversion.linear_obj_list[0], aa.m.MockLinearObjFuncList)
     assert isinstance(inversion, aa.InversionImagingMapping)
-    assert inversion.mapped_reconstructed_operated_data == pytest.approx(np.ones(9), 1.0e-4)
+    assert inversion.mapped_reconstructed_operated_data == pytest.approx(
+        np.ones(9), 1.0e-4
+    )
     assert inversion.reconstruction == pytest.approx(np.array([2.0]), 1.0e-4)
 
     # Overwrites use_sparse_operator to false.
@@ -38,7 +40,9 @@ def test__inversion_imaging__via_linear_obj_func_list(masked_imaging_7x7_no_blur
 
     assert isinstance(inversion.linear_obj_list[0], aa.m.MockLinearObjFuncList)
     assert isinstance(inversion, aa.InversionImagingMapping)
-    assert inversion.mapped_reconstructed_operated_data == pytest.approx(np.ones(9), 1.0e-4)
+    assert inversion.mapped_reconstructed_operated_data == pytest.approx(
+        np.ones(9), 1.0e-4
+    )
     assert inversion.reconstruction == pytest.approx(np.array([2.0]), 1.0e-4)
 
     # Works with multiple parameters
@@ -56,7 +60,9 @@ def test__inversion_imaging__via_linear_obj_func_list(masked_imaging_7x7_no_blur
     assert isinstance(inversion.linear_obj_list[0], aa.m.MockLinearObjFuncList)
     assert isinstance(inversion, aa.InversionImagingMapping)
     assert inversion.reconstruction == pytest.approx(np.array([0.0, 2.0]), abs=1.0e-4)
-    assert inversion.mapped_reconstructed_operated_data == pytest.approx(np.ones(9), 1.0e-4)
+    assert inversion.mapped_reconstructed_operated_data == pytest.approx(
+        np.ones(9), 1.0e-4
+    )
 
 
 def test__inversion_imaging__via_mapper(
@@ -78,7 +84,9 @@ def test__inversion_imaging__via_mapper(
     # assert inversion.log_det_curvature_reg_matrix_term == pytest.approx(
     #     4.609440907938719, 1.0e-4
     # )
-    assert inversion.mapped_reconstructed_operated_data == pytest.approx(np.ones(9), 1.0e-4)
+    assert inversion.mapped_reconstructed_operated_data == pytest.approx(
+        np.ones(9), 1.0e-4
+    )
 
     masked_imaging_7x7_no_blur_sparse_operator = (
         masked_imaging_7x7_no_blur.apply_sparse_operator_cpu()
@@ -93,7 +101,9 @@ def test__inversion_imaging__via_mapper(
     assert inversion.log_det_curvature_reg_matrix_term == pytest.approx(
         7.257175708246, 1.0e-4
     )
-    assert inversion.mapped_reconstructed_operated_data == pytest.approx(np.ones(9), 1.0e-4)
+    assert inversion.mapped_reconstructed_operated_data == pytest.approx(
+        np.ones(9), 1.0e-4
+    )
 
     inversion = aa.Inversion(
         dataset=masked_imaging_7x7_no_blur,
@@ -103,7 +113,9 @@ def test__inversion_imaging__via_mapper(
     assert isinstance(inversion.linear_obj_list[0], aa.MapperDelaunay)
     assert isinstance(inversion, aa.InversionImagingMapping)
     assert inversion.log_det_curvature_reg_matrix_term == pytest.approx(10.6674, 1.0e-4)
-    assert inversion.mapped_reconstructed_operated_data == pytest.approx(np.ones(9), 1.0e-4)
+    assert inversion.mapped_reconstructed_operated_data == pytest.approx(
+        np.ones(9), 1.0e-4
+    )
 
     inversion = aa.Inversion(
         dataset=masked_imaging_7x7_no_blur_sparse_operator,
@@ -112,7 +124,9 @@ def test__inversion_imaging__via_mapper(
 
     assert isinstance(inversion.linear_obj_list[0], aa.MapperDelaunay)
     assert inversion.log_det_curvature_reg_matrix_term == pytest.approx(10.6674, 1.0e-4)
-    assert inversion.mapped_reconstructed_operated_data == pytest.approx(np.ones(9), 1.0e-4)
+    assert inversion.mapped_reconstructed_operated_data == pytest.approx(
+        np.ones(9), 1.0e-4
+    )
 
 
 def test__inversion_imaging__via_regularizations(
@@ -141,7 +155,9 @@ def test__inversion_imaging__via_regularizations(
     assert inversion.log_det_curvature_reg_matrix_term == pytest.approx(
         10.66747, 1.0e-4
     )
-    assert inversion.mapped_reconstructed_operated_data == pytest.approx(np.ones(9), 1.0e-4)
+    assert inversion.mapped_reconstructed_operated_data == pytest.approx(
+        np.ones(9), 1.0e-4
+    )
 
     mapper = copy.copy(delaunay_mapper_9_3x3)
     mapper.regularization = regularization_adaptive_brightness
@@ -155,7 +171,9 @@ def test__inversion_imaging__via_regularizations(
     assert inversion.log_det_curvature_reg_matrix_term == pytest.approx(
         47.410169, 1.0e-4
     )
-    assert inversion.mapped_reconstructed_operated_data == pytest.approx(np.ones(9), 1.0e-4)
+    assert inversion.mapped_reconstructed_operated_data == pytest.approx(
+        np.ones(9), 1.0e-4
+    )
 
 
 def test__inversion_imaging__source_pixel_zeroed_indices(
@@ -210,7 +228,9 @@ def test__inversion_imaging__via_linear_obj_func_and_mapper(
         np.array([2.0]), 1.0e-4
     )
     assert inversion.reconstruction_dict[rectangular_mapper_7x7_3x3][0] < 1.0e-4
-    assert inversion.mapped_reconstructed_operated_data == pytest.approx(np.ones(9), 1.0e-4)
+    assert inversion.mapped_reconstructed_operated_data == pytest.approx(
+        np.ones(9), 1.0e-4
+    )
 
     masked_imaging_7x7_no_blur_sparse_operator = (
         masked_imaging_7x7_no_blur.apply_sparse_operator_cpu()
@@ -293,8 +313,11 @@ def test__inversion_imaging__compare_mapping_and_sparse_operator_values(
     assert inversion_sparse_operator.reconstruction == pytest.approx(
         inversion_mapping.reconstruction, 1.0e-4
     )
-    assert inversion_sparse_operator.mapped_reconstructed_operated_data.array == pytest.approx(
-        inversion_mapping.mapped_reconstructed_operated_data.array, 1.0e-4
+    assert (
+        inversion_sparse_operator.mapped_reconstructed_operated_data.array
+        == pytest.approx(
+            inversion_mapping.mapped_reconstructed_operated_data.array, 1.0e-4
+        )
     )
     assert inversion_sparse_operator.log_det_curvature_reg_matrix_term == pytest.approx(
         inversion_mapping.log_det_curvature_reg_matrix_term
@@ -546,7 +569,9 @@ def test__inversion_matrices__x2_mappers(
     assert inversion.mapped_reconstructed_operated_data_dict[delaunay_mapper_9_3x3][
         3
     ] == pytest.approx(0.49999704, 1.0e-4)
-    assert inversion.mapped_reconstructed_operated_data[4] == pytest.approx(0.99999998, 1.0e-4)
+    assert inversion.mapped_reconstructed_operated_data[4] == pytest.approx(
+        0.99999998, 1.0e-4
+    )
 
 
 def test__inversion_imaging__positive_only_solver(masked_imaging_7x7_no_blur):
@@ -566,7 +591,9 @@ def test__inversion_imaging__positive_only_solver(masked_imaging_7x7_no_blur):
 
     assert isinstance(inversion.linear_obj_list[0], aa.m.MockLinearObjFuncList)
     assert isinstance(inversion, aa.InversionImagingMapping)
-    assert inversion.mapped_reconstructed_operated_data == pytest.approx(np.ones(9), 1.0e-4)
+    assert inversion.mapped_reconstructed_operated_data == pytest.approx(
+        np.ones(9), 1.0e-4
+    )
     assert inversion.reconstruction == pytest.approx(np.array([2.0]), 1.0e-4)