simplify transformer tests

Author: Jammy2211

autoarray/dataset/imaging/simulator.py

@@ -33,7 +33,7 @@ def __init__(
         The simulation of an `Imaging` dataset uses the following steps:
 
         1) Receive as input a raw image of what the data looks like before any simulaiton process is applied.
-        2) Include dirrection due to the telescope optics by convolve the image with an input Point Spread
+        2) Include direction due to the telescope optics by convolve the image with an input Point Spread
            Function (PSF).
         3) Use input values of the background sky level in every pixel of the image to add the background sky to
            the PSF convolved image.

autoarray/inversion/pixelization/mappers/abstract.py

@@ -397,6 +397,9 @@ def extent_from(
         zoom_to_brightest: bool = True,
         zoom_percent: Optional[float] = None,
     ) -> Tuple[float, float, float, float]:
+
+        from autoarray.geometry import geometry_util
+
         if zoom_to_brightest and values is not None:
             if zoom_percent is None:
                 zoom_percent = conf.instance["visualize"]["general"]["zoom"][
@@ -408,8 +411,6 @@ def extent_from(
             true_indices = np.argwhere(fractional_bool)
             true_grid = self.source_plane_mesh_grid[true_indices]
 
-            from autoarray.geometry import geometry_util
-
             try:
                 return geometry_util.extent_symmetric_from(
                     extent=(
@@ -420,9 +421,13 @@ def extent_from(
                     )
                 )
             except ValueError:
-                return self.source_plane_mesh_grid.geometry.extent
+                return geometry_util.extent_symmetric_from(
+                    extent=self.source_plane_mesh_grid.geometry.extent
+                )
 
-        return self.source_plane_mesh_grid.geometry.extent
+        return geometry_util.extent_symmetric_from(
+            extent=self.source_plane_mesh_grid.geometry.extent
+        )
 
     def interpolated_array_from(
         self,

autoarray/operators/transformer_util.py

@@ -115,7 +115,7 @@ def visibilities_jit(image_1d, grid_radians, uv_wavelengths):
     return visibilities
 
 
-@numba_util.jit()
+# @numba_util.jit()
 def image_via_jit_from(n_pixels, grid_radians, uv_wavelengths, visibilities):
     image_1d = np.zeros(n_pixels)
 

autoarray/plot/mat_plot/abstract.py

@@ -108,7 +108,10 @@ def __init__(
         self.yticks = yticks or wb.YTicks(is_default=True)
         self.xticks = xticks or wb.XTicks(is_default=True)
 
-        self.title = title or wb.Title(is_default=True)
+        if title is not False:
+            self.title = title or wb.Title(is_default=True)
+        else:
+            self.title = False
         self.ylabel = ylabel or wb.YLabel(is_default=True)
         self.xlabel = xlabel or wb.XLabel(is_default=True)
 

autoarray/plot/mat_plot/two_d.py

@@ -333,7 +333,9 @@ def plot_array(
         else:
             title = auto_labels.title
 
-        self.title.set(auto_title=title, use_log10=self.use_log10)
+        if self.title is not False:
+            self.title.set(auto_title=title, use_log10=self.use_log10)
+
         self.ylabel.set()
         self.xlabel.set()
 
@@ -439,7 +441,8 @@ def plot_grid(
                 if colorbar is not None and self.colorbar_tickparams is not None:
                     self.colorbar_tickparams.set(cb=colorbar)
 
-        self.title.set(auto_title=auto_labels.title)
+        if self.title is not False:
+            self.title.set(auto_title=auto_labels.title)
         self.ylabel.set()
         self.xlabel.set()
 
@@ -583,7 +586,8 @@ def _plot_rectangular_mapper(
             shape_native=mapper.shape_native,
         )
 
-        self.title.set(auto_title=auto_labels.title)
+        if self.title is not False:
+            self.title.set(auto_title=auto_labels.title)
         self.tickparams.set()
         self.ylabel.set()
         self.xlabel.set()
@@ -668,7 +672,8 @@ def _plot_delaunay_mapper(
                 use_log10=self.use_log10,
             )
 
-        self.title.set(auto_title=auto_labels.title)
+        if self.title is not False:
+            self.title.set(auto_title=auto_labels.title)
         self.ylabel.set()
         self.xlabel.set()
 
@@ -751,7 +756,8 @@ def _plot_voronoi_mapper(
                 use_log10=self.use_log10,
             )
 
-        self.title.set(auto_title=auto_labels.title)
+        if self.title is not False:
+            self.title.set(auto_title=auto_labels.title)
         self.ylabel.set()
         self.xlabel.set()
 

autoarray/plot/wrap/base/units.py

@@ -15,7 +15,7 @@ def __init__(
         ticks_label: Optional[str] = None,
         colorbar_convert_factor: Optional[float] = None,
         colorbar_label: Optional[str] = None,
-        **kwargs
+        **kwargs,
     ):
         """
         This object controls the units of a plotted figure, and performs multiple tasks when making the plot:

autoarray/plot/wrap/one_d/yx_plot.py

@@ -72,7 +72,7 @@ def plot_y_vs_x(
                 #     marker="o",
                 fmt="o",
                 # ls=ls_errorbar,
-                **self.config_dict
+                **self.config_dict,
             )
             if plot_axis_type == "errorbar_logy":
                 plt.yscale("log")

autoarray/plot/wrap/two_d/delaunay_drawer.py

@@ -87,17 +87,18 @@ def draw_delaunay_pixels(
             cmap = plt.get_cmap(cmap.cmap)
 
             if colorbar is not None:
-                cb = colorbar.set_with_color_values(
-                    units=units,
-                    norm=norm,
-                    cmap=cmap,
-                    color_values=color_values,
-                    ax=ax,
-                    use_log10=use_log10,
-                )
-
-                if cb is not None and colorbar_tickparams is not None:
-                    colorbar_tickparams.set(cb=cb)
+                if colorbar is not False:
+                    cb = colorbar.set_with_color_values(
+                        units=units,
+                        norm=norm,
+                        cmap=cmap,
+                        color_values=color_values,
+                        ax=ax,
+                        use_log10=use_log10,
+                    )
+
+                    if cb is not None and colorbar_tickparams is not None:
+                        colorbar_tickparams.set(cb=cb)
 
         else:
 
@@ -115,5 +116,5 @@ def draw_delaunay_pixels(
             cmap=cmap,
             vmin=vmin,
             vmax=vmax,
-            **self.config_dict
+            **self.config_dict,
         )

autoarray/structures/arrays/kernel_2d.py

@@ -26,7 +26,7 @@ def __init__(
         normalize: bool = False,
         store_native: bool = False,
         *args,
-        **kwargs
+        **kwargs,
     ):
         """
         An array of values, which are paired to a uniform 2D mask of pixels. Each entry
@@ -91,7 +91,9 @@ def no_mask(
             pixel_scales=pixel_scales,
             origin=origin,
         )
-        return Kernel2D(values=values, mask=values.mask, header=header, normalize=normalize)
+        return Kernel2D(
+            values=values, mask=values.mask, header=header, normalize=normalize
+        )
 
     @classmethod
     def full(

autoarray/structures/plot/structure_plotters.py

@@ -210,5 +210,5 @@ def figure_1d(self):
             should_plot_grid=self.should_plot_grid,
             should_plot_zero=self.should_plot_zero,
             plot_axis_type_override=self.plot_axis_type,
-            **self.plot_yx_dict
+            **self.plot_yx_dict,
         )