Commit f915dce9 authored by Robert P. Goldman's avatar Robert P. Goldman Committed by Robert P. Goldman

Fix test bugs.

parent 94660a50
......@@ -277,4 +277,4 @@ class TestSaveLoad:
assert (value == ppc2[key]).all()
for key, value in ppcf.items():
assert (value == ppc2[key]).all()
assert (value == ppc2f[key]).all()
import pymc3 as pm
import pytest
from pymc3.distributions.posterior_predictive import _TraceDict
from numpy import array_equal
import numpy as np
from pymc3.backends.ndarray import point_list_to_multitrace
......@@ -22,7 +22,7 @@ def test_build_TraceDict():
dict = _TraceDict(multi_trace=trace)
assert isinstance(dict, _TraceDict)
assert len(dict) == 1000
assert array_equal(trace['mu'], dict['mu'])
np.testing.assert_array_equal(trace['mu'], dict['mu'])
assert set(trace.varnames) == set(dict.varnames) == set(["mu"])
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment