Skip to content

Commit d51d4d0

Browse files
committed
remove use of fixture
1 parent 9090e1f commit d51d4d0

File tree

1 file changed

+4
-9
lines changed

1 file changed

+4
-9
lines changed

tests/test_model_graph.py

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -536,14 +536,6 @@ def simple_model() -> pm.Model:
536536
return model
537537

538538

539-
@pytest.fixture(scope="module")
540-
def variable_with_space():
541-
with pm.Model() as model:
542-
pm.Normal("plant growth")
543-
544-
return model
545-
546-
547539
def test_unknown_node_type(simple_model):
548540
with pytest.raises(ValueError, match="Node formatters must be of type NodeType."):
549541
model_to_graphviz(simple_model, node_formatters={"Unknown Node Type": "dummy"})
@@ -662,7 +654,10 @@ def test_model_to_mermaid(simple_model):
662654
assert model_to_mermaid(simple_model) == expected_mermaid_string.strip()
663655

664656

665-
def test_model_to_mermaid_with_variable_with_space(variable_with_space):
657+
def test_model_to_mermaid_with_variable_with_space():
658+
with pm.Model() as variable_with_space:
659+
pm.Normal("plant growth")
660+
666661
expected_mermaid_string = dedent("""
667662
graph TD
668663
%% Nodes:

0 commit comments

Comments
 (0)