From 55b5a4432af187960a544dd406d4d58b06b4c831 Mon Sep 17 00:00:00 2001 From: Jan Kwakkel Date: Mon, 4 Dec 2023 19:05:31 +0100 Subject: [PATCH] Update test_prim.py --- test/test_analysis/test_prim.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/test/test_analysis/test_prim.py b/test/test_analysis/test_prim.py index ffd096af6..bf047491c 100644 --- 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):