fix coverage

Author: BatyLeo

src/Argmax2D/Argmax2D.jl

@@ -103,16 +103,13 @@ $TYPEDSIGNATURES
 Plot the data sample for the [`Argmax2DBenchmark`](@ref).
 """
 function Utils.plot_data(
-    ::Argmax2DBenchmark,
+    bench::Argmax2DBenchmark,
     sample::DataSample;
     instance=sample.instance,
     θ=sample.θ_true,
     kwargs...,
 )
-    pl = init_plot()
-    plot_polytope!(pl, instance)
-    plot_objective!(pl, θ)
-    return plot_maximizer!(pl, θ, instance, maximizer)
+    return Utils.plot_data(bench; instance, θ, kwargs...)
 end
 
 export Argmax2DBenchmark

test/argmax.jl

@@ -14,6 +14,9 @@
     model = generate_statistical_model(b)
     maximizer = generate_maximizer(b)
 
+    gap = compute_gap(b, dataset, model, maximizer)
+    @test gap >= 0
+
     for (i, sample) in enumerate(dataset)
         (; x, θ_true, y_true) = sample
         @test size(x) == (nb_features, instance_dim)

test/argmax_2d.jl

@@ -13,6 +13,9 @@
     model = generate_statistical_model(b)
     maximizer = generate_maximizer(b)
 
+    gap = compute_gap(b, dataset, model, maximizer)
+    @test gap >= 0
+
     # Test plot_data
     figure = plot_data(b, dataset[1])
     @test figure isa Plots.Plot