| | """Tests for network builder."""
|
| |
|
| | import pytest
|
| | import sys, os
|
| | sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
|
| |
|
| | import neurocore as nc
|
| | from neurocore.exceptions import (
|
| | NetworkTooLargeError, WeightOutOfRangeError, NeurocoreError,
|
| | )
|
| | from neurocore.constants import MAX_CORES, NEURONS_PER_CORE
|
| |
|
| |
|
| | class TestPopulation:
|
| | def test_create_population(self):
|
| | net = nc.Network()
|
| | pop = net.population(64, label="test")
|
| | assert pop.size == 64
|
| | assert pop.label == "test"
|
| | assert pop.id == 0
|
| |
|
| | def test_population_params_dict(self):
|
| | net = nc.Network()
|
| | pop = net.population(16, params={"threshold": 800, "leak": 5})
|
| | assert pop.params.threshold == 800
|
| | assert pop.params.leak == 5
|
| | assert pop.params.resting == 0
|
| |
|
| | def test_population_invalid_param(self):
|
| | net = nc.Network()
|
| | with pytest.raises(ValueError, match="Unknown neuron parameter"):
|
| | net.population(16, params={"bogus": 42})
|
| |
|
| | def test_population_zero_size(self):
|
| | net = nc.Network()
|
| | with pytest.raises(ValueError, match="positive"):
|
| | net.population(0)
|
| |
|
| | def test_population_slicing(self):
|
| | net = nc.Network()
|
| | pop = net.population(32)
|
| | s = pop[:8]
|
| | assert len(s) == 8
|
| | assert s.indices == list(range(8))
|
| |
|
| | def test_population_single_index(self):
|
| | net = nc.Network()
|
| | pop = net.population(10)
|
| | s = pop[5]
|
| | assert len(s) == 1
|
| | assert s.indices == [5]
|
| |
|
| | def test_population_negative_index(self):
|
| | net = nc.Network()
|
| | pop = net.population(10)
|
| | s = pop[-1]
|
| | assert s.indices == [9]
|
| |
|
| | def test_population_index_out_of_range(self):
|
| | net = nc.Network()
|
| | pop = net.population(10)
|
| | with pytest.raises(IndexError):
|
| | pop[10]
|
| |
|
| |
|
| | class TestConnection:
|
| | def test_create_connection(self):
|
| | net = nc.Network()
|
| | a = net.population(8)
|
| | b = net.population(8)
|
| | conn = net.connect(a, b, topology="all_to_all", weight=200)
|
| | assert conn.source is a
|
| | assert conn.target is b
|
| | assert conn.weight == 200
|
| |
|
| | def test_weight_out_of_range(self):
|
| | net = nc.Network()
|
| | a = net.population(8)
|
| | b = net.population(8)
|
| | with pytest.raises(WeightOutOfRangeError):
|
| | net.connect(a, b, weight=40000)
|
| |
|
| | def test_invalid_compartment(self):
|
| | net = nc.Network()
|
| | a = net.population(8)
|
| | b = net.population(8)
|
| | with pytest.raises(ValueError, match="Compartment"):
|
| | net.connect(a, b, compartment=5)
|
| |
|
| | def test_negative_weight(self):
|
| | net = nc.Network()
|
| | a = net.population(8)
|
| | b = net.population(8)
|
| | conn = net.connect(a, b, weight=-300)
|
| | assert conn.weight == -300
|
| |
|
| |
|
| | class TestNetwork:
|
| | def test_total_neurons(self):
|
| | net = nc.Network()
|
| | net.population(64)
|
| | net.population(16)
|
| | assert net.total_neurons() == 80
|
| |
|
| | def test_validate_ok(self, small_network):
|
| | net, _, _ = small_network
|
| | warnings = net.validate()
|
| | assert warnings == []
|
| |
|
| | def test_validate_too_large(self):
|
| | net = nc.Network()
|
| |
|
| | net.population(MAX_CORES * NEURONS_PER_CORE + 1)
|
| | with pytest.raises(NetworkTooLargeError):
|
| | net.validate()
|
| |
|
| | def test_validate_empty(self):
|
| | net = nc.Network()
|
| | warnings = net.validate()
|
| | assert "no neurons" in warnings[0].lower()
|
| |
|
| | def test_repr(self):
|
| | net = nc.Network()
|
| | net.population(10)
|
| | assert "neurons=10" in repr(net)
|
| |
|
