From 175e04b331e2a9ef46f09b3b05d9075d28befeeb Mon Sep 17 00:00:00 2001
From: Carleton Coffrin <carleton@coffrin.com>
Date: Mon, 25 Mar 2024 08:04:59 -0700
Subject: [PATCH] add test of plotting features

---
 test/plot.jl | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/test/plot.jl b/test/plot.jl
index d803c7a..4de21a5 100644
--- 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