Spaces:
Paused
Paused
| """Bounded asynchronous storage tests.""" | |
| from __future__ import annotations | |
| from pathlib import Path | |
| from src.storage.async_runtime import AsyncStorageConfig, AsyncStorageSession | |
| from src.storage.delta_journal import DeltaJournal, DeltaType | |
| from src.storage.runtime import StorageRuntimeConfig | |
| from tests.test_storage_runtime import FakeNetwork, FakeNeuron, FakeStep, FakeSynapse | |
| def test_async_session_writes_and_reports_telemetry(tmp_path: Path) -> None: | |
| network = FakeNetwork( | |
| dimensions=(10, 10, 10, 10, 10), | |
| current_tick=0, | |
| neurons={1: FakeNeuron(), 2: FakeNeuron()}, | |
| synapses={1: [FakeSynapse(2, 0.2, 1)], 2: []}, | |
| ) | |
| runtime = StorageRuntimeConfig( | |
| snapshot_path=tmp_path / "async.b5d", | |
| journal_path=tmp_path / "async.b5d.journal", | |
| commit_interval_ticks=1, | |
| ) | |
| session = AsyncStorageSession( | |
| network, # type: ignore[arg-type] | |
| runtime, | |
| AsyncStorageConfig(queue_size=8), | |
| ) | |
| session.start() | |
| network.current_tick = 1 | |
| network.neurons[1].v = -40.0 | |
| network.neurons[1].spike_counter = 1 | |
| network.neurons[1].last_spike_tick = 1 | |
| for hook in tuple(network.hooks): | |
| hook(FakeStep(1, (1,))) | |
| session.close() | |
| telemetry = session.telemetry | |
| assert telemetry.batches_enqueued == 1 | |
| assert telemetry.batches_written == 1 | |
| assert telemetry.deltas_written >= 2 | |
| assert telemetry.dropped_batches == 0 | |
| assert not telemetry.worker_failed | |
| with DeltaJournal(runtime.journal_path) as journal: | |
| kinds = {entry.delta_type for entry in journal.iter_committed_entries()} | |
| assert DeltaType.NEURON_STATE in kinds | |
| assert DeltaType.SPIKE_EVENT in kinds | |