File size: 2,509 Bytes
e4cdd5f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 | """Tests for analysis functions."""
import pytest
import sys, os
sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
import numpy as np
from neurocore.result import RunResult
from neurocore import analysis
@pytest.fixture
def mock_result():
"""A RunResult with known spike data."""
return RunResult(
total_spikes=10,
timesteps=100,
spike_trains={
0: [5, 15, 25, 35, 45],
1: [10, 20, 30],
2: [50, 60],
},
placement=None,
backend="simulator",
)
class TestFiringRates:
def test_per_neuron(self, mock_result):
rates = analysis.firing_rates(mock_result)
assert rates[0] == pytest.approx(5 / 100)
assert rates[1] == pytest.approx(3 / 100)
assert rates[2] == pytest.approx(2 / 100)
def test_hardware_aggregate(self):
result = RunResult(
total_spikes=500, timesteps=100,
spike_trains={}, placement=None, backend="chip",
)
rates = analysis.firing_rates(result)
assert rates["aggregate"] == pytest.approx(5.0)
class TestSpikeCountTimeseries:
def test_basic(self, mock_result):
ts = analysis.spike_count_timeseries(mock_result, bin_size=10)
assert len(ts) == 10
# Bin 0 (t=0-9): spike at t=5 -> 1
assert ts[0] == 1
# Bin 1 (t=10-19): spikes at t=10, 15 -> 2
assert ts[1] == 2
def test_empty(self):
result = RunResult(0, 100, {}, None, "chip")
ts = analysis.spike_count_timeseries(result)
assert len(ts) == 0
class TestISIHistogram:
def test_basic(self, mock_result):
counts, edges = analysis.isi_histogram(mock_result, bins=5)
assert len(counts) == 5
assert counts.sum() > 0
def test_empty(self):
result = RunResult(0, 100, {}, None, "simulator")
counts, edges = analysis.isi_histogram(result)
assert len(counts) == 0
class TestRasterPlot:
def test_raster_no_display(self, mock_result):
"""Test raster plot generates without error (non-interactive)."""
import matplotlib
matplotlib.use("Agg")
fig = analysis.raster_plot(mock_result, show=False)
assert fig is not None
def test_raster_hardware_fails(self):
result = RunResult(100, 50, {}, None, "chip")
with pytest.raises(Exception):
result.raster_plot()
|