mirror of
https://github.com/p2p-ld/nwb-linkml.git
synced 2024-11-14 02:34:28 +00:00
35 lines
1 KiB
Python
35 lines
1 KiB
Python
from typing import Tuple
|
|
|
|
import numpy as np
|
|
import pytest
|
|
|
|
# FIXME: Make this just be the output of the provider by patching into import machinery
|
|
from nwb_linkml.models.pydantic.core.v2_7_0.namespace import (
|
|
ElectricalSeries,
|
|
ElectrodeGroup,
|
|
NWBFileGeneralExtracellularEphysElectrodes,
|
|
)
|
|
|
|
|
|
@pytest.fixture()
|
|
def electrical_series() -> Tuple["ElectricalSeries", "NWBFileGeneralExtracellularEphysElectrodes"]:
|
|
"""
|
|
Demo electrical series with adjoining electrodes
|
|
"""
|
|
n_electrodes = 5
|
|
n_times = 100
|
|
data = np.arange(0, n_electrodes * n_times).reshape(n_times, n_electrodes)
|
|
timestamps = np.linspace(0, 1, n_times)
|
|
|
|
# electrode group is the physical description of the electrodes
|
|
electrode_group = ElectrodeGroup(
|
|
name="GroupA",
|
|
)
|
|
|
|
# make electrodes tables
|
|
electrodes = NWBFileGeneralExtracellularEphysElectrodes(
|
|
id=np.arange(0, n_electrodes),
|
|
x=np.arange(0, n_electrodes),
|
|
y=np.arange(n_electrodes, n_electrodes * 2),
|
|
group=[electrode_group] * n_electrodes,
|
|
)
|