add test of plotting features

lanl-ansi · Mar 25, 2024 · 175e04b · 175e04b
1 parent c193593
commit 175e04b
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test/plot.jl b/test/plot.jl
@@ -80,6 +80,16 @@ end
         @test true
     end
 
+    @testset "two qubit plot, num_states" begin
+        plt = plot_states(ρ,num_states=2)
+        @test true
+    end
+
+    @testset "two qubit plot, num_states" begin
+        plt = plot_states(ρ,ising_model=ising,energy_levels=1)
+        @test true
+    end
+
     @testset "two qubit plot, numeric sorting, with kwargs" begin
         plt = plot_states(ρ,order=:numeric,xlabel="x")
         @test true