more tests

Author: BatyLeo

test/vsp.jl

@@ -25,12 +25,26 @@
     # Test labeled stochastic dataset with SAA policy
     # N instances, each with K scenarios → N labeled samples
     saa_dataset = generate_dataset(
-        b, N; nb_scenarios=K, seed=0, rng=StableRNG(0), target_policy=policies.saa.policy
+        b, N; nb_scenarios=K, seed=0, rng=StableRNG(0), target_policy=policies.saa
     )
     @test length(saa_dataset) == N
     @test hasproperty(saa_dataset[1].extra, :scenarios)
     @test saa_dataset[1].extra.scenarios isa Vector{VSPScenario}
     @test length(saa_dataset[1].extra.scenarios) == K
+    det_dataset = generate_dataset(
+        b, N; nb_scenarios=K, seed=0, rng=StableRNG(0), target_policy=policies.deterministic
+    )
+    @test length(det_dataset) == N
+    @test hasproperty(det_dataset[1].extra, :scenarios)
+    @test det_dataset[1].extra.scenarios isa Vector{VSPScenario}
+    @test length(det_dataset[1].extra.scenarios) == K
+    ls_dataset = generate_dataset(
+        b, N; nb_scenarios=K, seed=0, rng=StableRNG(0), target_policy=policies.local_search
+    )
+    @test length(ls_dataset) == N
+    @test hasproperty(ls_dataset[1].extra, :scenarios)
+    @test ls_dataset[1].extra.scenarios isa Vector{VSPScenario}
+    @test length(ls_dataset[1].extra.scenarios) == K
 
     # Plots work unchanged
     figure_1 = plot_instance(b, saa_dataset[1])