Update test_prim.py

quaquel · Dec 4, 2023 · 55b5a44 · 55b5a44
1 parent 1df7de3
commit 55b5a44
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/test/test_analysis/test_prim.py b/test/test_analysis/test_prim.py
@@ -8,6 +8,7 @@
 
 import numpy as np
 import pandas as pd
+import matplotlib.pyplot as plt
 
 from ema_workbench.analysis import prim
 from ema_workbench.analysis.prim import PrimBox
@@ -69,9 +70,19 @@ def test_inspect(self):
         box.inspect(1)
         box.inspect()
         box.inspect(style="graph")
+        box.inspect(style="data")
 
         box.inspect([0, 1])
 
+        fig, axes = plt.subplots(2)
+        box.inspect([0, 1], ax=axes, style="graph")
+
+        fig, ax = plt.subplots()
+        box.inspect(0, ax=ax, style="graph")
+
+        with pytest.raises(ValueError):
+            fig, axes = plt.subplots(3)
+            box.inspect([0, 1], ax=axes, style="graph")
         with pytest.raises(ValueError):
             box.inspect(style="some unknown style")
         with pytest.raises(TypeError):