Fixed test for fake_treatments

dscolby · Oct 30, 2023 · bb08388 · bb08388
1 parent 2c3841b
commit bb08388
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/test_model_validation.jl b/test/test_model_validation.jl
@@ -79,10 +79,11 @@ end
     for vec in jenks(collect(1:10), 5)
         @test !isempty(vec)
     end
+    setdiff
 end
 
 @testset "Counterfactual Consistency" begin
     @test length(fake_treatments(test_outcomes)) == length(test_outcomes)
-    @test sort(unique(fake_treatments(test_outcomes))) == [1, 2, 3, 4]
+    @test setdiff(Set(sort(unique(fake_treatments(test_outcomes)))), [1, 2, 3, 4]) == Set()
     @test counterfactualconsistency(g_computer) isa Real
 end