plot_data tests

Author: BatyLeo

test/argmax_2d.jl

@@ -1,5 +1,6 @@
 @testitem "Argmax2D" begin
     using DecisionFocusedLearningBenchmarks
+    using Plots
 
     nb_features = 5
     b = Argmax2DBenchmark(; nb_features=nb_features)
@@ -12,6 +13,10 @@
     model = generate_statistical_model(b)
     maximizer = generate_maximizer(b)
 
+    # Test plot_data
+    figure = plot_data(b, dataset[1])
+    @test figure isa Plots.Plot
+
     for (i, sample) in enumerate(dataset)
         (; x, θ_true, y_true, instance) = sample
         @test length(x) == nb_features