Skip to content

Commit a28cc07

Browse files
committed
Merge branch 'probegroup-get-slice' of github.com:alejoe91/probeinterface into probegroup-get-slice
2 parents 33fb7fc + 74c31a9 commit a28cc07

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

tests/test_probegroup.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55

66
import numpy as np
77

8+
89
@pytest.fixture
910
def probegroup():
1011
"""Fixture: a ProbeGroup with 3 probes, each with device channel indices set."""
@@ -19,6 +20,7 @@ def probegroup():
1920
nchan += n
2021
return probegroup
2122

23+
2224
def test_probegroup(probegroup):
2325
indices = probegroup.get_global_device_channel_indices()
2426

@@ -105,6 +107,7 @@ def test_set_contact_ids_rejects_wrong_size():
105107
with pytest.raises(ValueError, match="do not have the same size"):
106108
probe.set_contact_ids(["a", "b", "c"])
107109

110+
108111
# ── get_global_contact_positions() tests ────────────────────────────────────
109112

110113

@@ -124,7 +127,9 @@ def test_get_global_contact_positions_matches_per_probe(probegroup):
124127

125128
def test_get_global_contact_positions_single_probe(probegroup):
126129
pos = probegroup.get_global_contact_positions()
127-
np.testing.assert_array_equal(pos[: probegroup.probes[0].get_contact_count()], probegroup.probes[0].contact_positions)
130+
np.testing.assert_array_equal(
131+
pos[: probegroup.probes[0].get_contact_count()], probegroup.probes[0].contact_positions
132+
)
128133

129134

130135
def test_get_global_contact_positions_3d():
@@ -148,6 +153,7 @@ def test_get_global_contact_positions_reflects_move():
148153
pos = pg.get_global_contact_positions()
149154
np.testing.assert_array_equal(pos, original_pos + np.array([50, 60]))
150155

156+
151157
# ── copy() tests ────────────────────────────────────────────────────────────
152158

153159

0 commit comments

Comments
 (0)