Python model_to_graphviz Examples

Example #1

File: test_model_graph.py Project: fonnesbeck/pymc3

    def test_graphviz(self):
        # just make sure everything runs without error

        g = self.model_graph.make_graph()
        for key in self.compute_graph:
            assert key in g.source
        g = model_to_graphviz(self.model)
        for key in self.compute_graph:
            assert key in g.source

Example #2

File: test_model_graph.py Project: fonnesbeck/pymc3

 def test_checks_formatting(self):
     with pytest.warns(None):
         model_to_graphviz(self.model, formatting="plain")
     with pytest.raises(ValueError, match="Unsupported formatting"):
         model_to_graphviz(self.model, formatting="latex")
     with pytest.warns(UserWarning, match="currently not supported"):
         model_to_graphviz(self.model, formatting="plain_with_params")