Check for error loading bad trace directory.

pymc-devs · Jun 27, 2019 · 783c4a7 · 783c4a7
1 parent 619363b
commit 783c4a7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pymc3/tests/test_ndarray_backend.py b/pymc3/tests/test_ndarray_backend.py
@@ -213,6 +213,11 @@ def test_save_and_load(self, tmpdir_factory):
         for var in ('x', 'z'):
             assert (self.trace[var] == trace2[var]).all()
 
+    def test_bad_load(self, tmpdir_factory):
+        directory = str(tmpdir_factory.mktemp('data'))
+        with pytest.raises(Exception):
+            pm.load_trace(directory, model=TestSaveLoad.model())
+
     def test_sample_posterior_predictive(self, tmpdir_factory):
         directory = str(tmpdir_factory.mktemp('data'))
         save_dir = pm.save_trace(self.trace, directory, overwrite=True)