from os import path

import pytest

import autolens as al
import autolens.plot as aplt

directory = path.dirname(path.realpath(__file__))


@pytest.fixture(name="plot_path")
def make_fit_imaging_plotter_setup():
    return path.join(
        "{}".format(path.dirname(path.realpath(__file__))), "files", "plots", "fit"
    )


# def test__subhalo_detection_sub_plot(
#     fit_imaging_x2_plane_7x7,
#     fit_imaging_x2_plane_inversion_7x7,
#     include_2d_all,
#     plot_path,
#     plot_patch,
# ):
#     arr = al.Array2D.manual_native(array=[[1.0, 2.0], [3.0, 4.0]], pixel_scales=1.0)
#
#     subhalo_plotter = aplt.SubhaloPlotter(
#         include_2d=include_2d_all,
#         mat_plot_2d=aplt.MatPlot2D(output=aplt.Output(path=plot_path, format="png")),
#     )
#
#     subhalo_plotter.subplot_detection_imaging(
#         fit_imaging_detect=fit_imaging_x2_plane_7x7, detection_array=arr, mass_array=arr
#     )
#
#     assert path.join(plot_path, "subplot_detection_imaging.png") in plot_patch.paths
#
#     subhalo_plotter.subplot_detection_imaging(
#         fit_imaging_detect=fit_imaging_x2_plane_inversion_7x7,
#         detection_array=arr,
#         mass_array=arr,
#     )
#
#     assert path.join(plot_path, "subplot_detection_imaging.png") in plot_patch.paths
#
#
# def test__subhalo_detection_fits(
#     fit_imaging_x2_plane_7x7,
#     fit_imaging_x2_plane_inversion_7x7,
#     include_2d_all,
#     plot_path,
#     plot_patch,
# ):
#
#     subhalo_plotter = aplt.SubhaloPlotter(
#         include_2d=include_2d_all,
#         mat_plot_2d=aplt.MatPlot2D(output=aplt.Output(path=plot_path, format="png")),
#     )
#
#     subhalo_plotter.subplot_detection_fits(
#         fit_imaging_before=fit_imaging_x2_plane_7x7,
#         fit_imaging_detect=fit_imaging_x2_plane_7x7,
#     )
#
#     assert path.join(plot_path, "subplot_detection_fits.png") in plot_patch.paths
#
#     subhalo_plotter.subplot_detection_fits(
#         fit_imaging_before=fit_imaging_x2_plane_inversion_7x7,
#         fit_imaging_detect=fit_imaging_x2_plane_inversion_7x7,
#     )
#
#     assert path.join(plot_path, "subplot_detection_fits.png") in plot_patch.paths
