def test_print_neuron_group_map(self):
map = api.MapSelector()
map.source = api.Population(population="Foo", count=42)
map.pattern = api.IndexTemplate(index="i")
self.assertEqual(
print_neuron_group(map),
"nrp.map_neurons(range(0, 42), lambda i: nrp.brain.Foo[i])")
def test_print_neurons_population(self):
pop = api.Population()
pop.population = "Foo"
pop.count = 42
self.assertEqual(get_neurons_index(pop), None)
self.assertEqual(print_neurons(pop), "Foo")
self.assertEqual(print_neurons(pop, "nrp.brain."), "nrp.brain.Foo")
def test_print_neuron_group_chain(self):
chain = api.ChainSelector()
chain.connectors.append(api.ChainSelector())
self.assertEqual(print_neuron_group(chain),
"nrp.chain_neurons(nrp.chain_neurons())")
chain = api.ChainSelector()
chain.neurons.append(api.Population(population="Foo", count=42))
self.assertEqual(print_neuron_group(chain),
"nrp.chain_neurons(nrp.brain.Foo)")
chain.neurons.append(api.Population(population="Bar", count=23))
self.assertEqual(print_neuron_group(chain),
"nrp.chain_neurons(nrp.brain.Foo, nrp.brain.Bar)")
chain.connectors.append(api.ChainSelector())
self.assertEqual(
print_neuron_group(chain),
"nrp.chain_neurons(nrp.brain.Foo, nrp.brain.Bar, nrp.chain_neurons())"
)
def test_get_all_neurons_as_dict_range(self):
populations = [api.Range(population='foo', from_=0, to=2)]
self.assertEqual(get_all_neurons_as_dict(populations),
{'foo': slice(0, 2)})
populations = [api.List(population='foo', element=[1, 3])]
self.assertEqual(get_all_neurons_as_dict(populations), {'foo': [1, 3]})
populations = [api.Population(population='foo', count=5)]
self.assertEqual(get_all_neurons_as_dict(populations), {'foo': 'foo'})
self.assertRaises(Exception, get_all_neurons_as_dict, ['foo'])
def test_get_neuron_count(self):
self.assertEqual(
3, get_neuron_count(api.List(population="pop", element=[1, 2, 3])))
self.assertEqual(1, get_neuron_count(api.Index(population="pop")))
self.assertEqual(
4, get_neuron_count(api.Range(population="pop", from_=3, to=7)))
self.assertEqual(
2,
get_neuron_count(api.Range(population="pop", from_=0, to=5,
step=2)))
self.assertEqual(
42, get_neuron_count(api.Population(population="pop", count=42)))
self.assertRaises(Exception, get_neuron_count, 'foo')
def test_get_collection_source_population(self):
pop = api.Population()
pop.population = "Foo"
pop.count = 42
self.assertEqual(get_collection_source(pop), "range(0, 42)")