fix cov test

arviz-devs · Feb 21, 2020 · 0191a8d · 0191a8d
1 parent 52ffbf2
commit 0191a8d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/arviz/tests/base_tests/test_plots_matplotlib.py b/arviz/tests/base_tests/test_plots_matplotlib.py
@@ -294,8 +294,9 @@ def test_plot_kde(continuous_model, kwargs):
     assert axes
 
 
-@pytest.mark.parametrize("x", [np.random.randn(8), np.random.randn(8, 8), np.random.randn(8, 8, 8)])
-def test_cov(x):
+@pytest.mark.parametrize("shape", [(8,), (8, 8), (8, 8, 8)])
+def test_cov(shape):
+    x = np.random.randn(*shape)
     if x.ndim <= 2:
         assert np.allclose(_cov(x), np.cov(x))
     else: