55
66import numpy as np
77
8+
89@pytest .fixture
910def 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+
2224def 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
125128def 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
130135def 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