def test_probe_all_kwargs(): net = make_net() probe_all(net, recursive=True, sample_every=0.1, seed=10) for probe in net.probes: assert probe.sample_every == 0.1 assert probe.seed == 10
def test_probe_all_options(recursive): net = make_net() probe_all(net, recursive=recursive, probe_options={ nengo.Ensemble: ['decoded_output']}) if recursive: assert(len(net.probes) == 2) else: assert(len(net.probes) == 1)
def test_probe_all_recursive(recursive): net = make_net() probes = probe_all(net, recursive=recursive) assert(len(probes[net.ens1]) == len(net.ens1.probeable)) assert(len(probes[net.ens1.neurons]) == len(net.ens1.neurons.probeable)) assert(len(probes[net.node1]) == len(net.node1.probeable)) assert(len(probes[net.conn]) == len(net.conn.probeable)) for lr in net.conn.learning_rule: assert(len(probes[lr]) == len(lr.probeable)) if recursive: assert(len(probes) == 8) assert(len(probes[net.ens2]) == len(net.ens2.probeable)) assert(len(probes[net.node2]) == len(net.node2.probeable)) else: assert(len(probes) == 5) assert net.ens2 not in probes assert net.ens2.neurons not in probes assert net.node2 not in probes
def test_probe_all_recursive(recursive): net = make_net() probes = probe_all(net, recursive=recursive) assert len(probes[net.ens1]) == len(net.ens1.probeable) # TODO: remove `set` when duplicate "output" bug fixed assert len(probes[net.ens1.neurons]) == len(set(net.ens1.neurons.probeable)) assert len(probes[net.node1]) == len(net.node1.probeable) assert len(probes[net.conn]) == len(net.conn.probeable) for lr in net.conn.learning_rule: assert len(probes[lr]) == len(lr.probeable) if recursive: assert len(probes) == 8 assert len(probes[net.ens2]) == len(net.ens2.probeable) assert len(probes[net.node2]) == len(net.node2.probeable) else: assert len(probes) == 5 assert net.ens2 not in probes assert net.ens2.neurons not in probes assert net.node2 not in probes