def test_create_dot(): """Constructing a .dot file for a model""" model = nengo.Model() with model: D = 3 input = nengo.Node([1]*D, label='input') a = nengo.networks.EnsembleArray(50, D, label='a') b = nengo.networks.EnsembleArray(50, D, label='b') output = nengo.Node(None, size_in=D, label='output') nengo.Connection(input, a.input, filter=0.01) nengo.Connection(a.output, b.input, filter=0.01) nengo.Connection(b.output, b.input, filter=0.01, transform=0.9) nengo.Connection(a.output, a.input, filter=0.01, transform=np.ones((D, D))) nengo.Connection(b.output, output, filter=0.01) dot = generate_graphviz(model.objs.values(), model.connections.values()) assert len(dot.splitlines()) == 31 # not sure what else to check here objs, conns = remove_passthrough_nodes(model.objs.values(), model.connections.values()) dot = generate_graphviz(objs, conns) assert len(dot.splitlines()) == 27
def test_create_dot(): """Constructing a .dot file for a model""" model = nengo.Network() with model: D = 3 input = nengo.Node([1] * D, label='input') a = nengo.networks.EnsembleArray(50, D, label='a') b = nengo.networks.EnsembleArray(50, D, label='b') output = nengo.Node(None, size_in=D, label='output') nengo.Connection(input, a.input, synapse=0.01) nengo.Connection(a.output, b.input, synapse=0.01) nengo.Connection(b.output, b.input, synapse=0.01, transform=0.9) nengo.Connection(a.output, a.input, synapse=0.01, transform=np.ones((D, D))) nengo.Connection(b.output, output, synapse=0.01) dot = generate_graphviz(*objs_and_connections(model)) assert len(dot.splitlines()) == 31 # not sure what else to check here dot = generate_graphviz(*remove_passthrough_nodes( *objs_and_connections(model))) assert len(dot.splitlines()) == 27
def test_moved_error(): with pytest.raises(MovedError) as excinfo: generate_graphviz() assert str(excinfo.value) == "This feature has been moved to nengo_extras.graphviz" check_tb_entries( excinfo.traceback, [ ("test_moved_error", "generate_graphviz()"), ("generate_graphviz", 'raise MovedError(location="nengo_extras.graphviz")'), ], )
def test_graphviz_moved(): with pytest.raises(MovedError): generate_graphviz()
def test_graphviz_moved(): with pytest.raises(MovedError): generate_graphviz()