diff options
| author | Cathy Yeh <cathy@driver.xyz> | 2017-12-03 19:16:32 -0800 |
|---|---|---|
| committer | Cathy Yeh <cathy@driver.xyz> | 2017-12-03 20:35:30 -0800 |
| commit | 8dc7ae89677fca16ee974a30cff8c4df53c955ce (patch) | |
| tree | 6b021dcd7902a2952cc97872f6200469b7dab51b /tests | |
| parent | e5937060658f7e8ac484e5489f2b35b4ecb96d35 (diff) | |
| download | beliefs-8dc7ae89677fca16ee974a30cff8c4df53c955ce.tar.gz beliefs-8dc7ae89677fca16ee974a30cff8c4df53c955ce.tar.bz2 beliefs-8dc7ae89677fca16ee974a30cff8c4df53c955ce.zip | |
PR comments
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_belief_propagation.py | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/tests/test_belief_propagation.py b/tests/test_belief_propagation.py index 264ddae..5c5a612 100644 --- a/tests/test_belief_propagation.py +++ b/tests/test_belief_propagation.py @@ -3,8 +3,10 @@ import pytest from pytest import approx from beliefs.inference.belief_propagation import BeliefPropagation, ConflictingEvidenceError -from beliefs.models.beliefupdate.BeliefUpdateNodeModel import BeliefUpdateNodeModel -from beliefs.models.beliefupdate.BernoulliOrNode import BernoulliOrNode +from beliefs.models.belief_update_node_model import ( + BeliefUpdateNodeModel, + BernoulliOrNode +) @pytest.fixture(scope='module') @@ -37,17 +39,17 @@ def many_parents_edges(): @pytest.fixture(scope='function') def four_node_model(edges_four_nodes): - return BeliefUpdateNodeModel.from_edges(edges_four_nodes, BernoulliOrNode) + return BeliefUpdateNodeModel.init_from_edges(edges_four_nodes, BernoulliOrNode) @pytest.fixture(scope='function') def simple_model(simple_edges): - return BeliefUpdateNodeModel.from_edges(simple_edges, BernoulliOrNode) + return BeliefUpdateNodeModel.init_from_edges(simple_edges, BernoulliOrNode) @pytest.fixture(scope='function') def many_parents_model(many_parents_edges): - return BeliefUpdateNodeModel.from_edges(many_parents_edges, BernoulliOrNode) + return BeliefUpdateNodeModel.init_from_edges(many_parents_edges, BernoulliOrNode) @pytest.fixture(scope='function') |