From 8e514e502f2a36f4173ae82c1cf7d2ae66aa03b0 Mon Sep 17 00:00:00 2001 From: Heberto Mayorquin Date: Thu, 23 Apr 2026 20:48:14 -0600 Subject: [PATCH] error --- src/probeinterface/probe.py | 2 +- tests/test_probe.py | 13 +++++++++++++ 2 files changed, 14 insertions(+), 1 deletion(-) diff --git a/src/probeinterface/probe.py b/src/probeinterface/probe.py index c19ddd65..d260a17f 100644 --- a/src/probeinterface/probe.py +++ b/src/probeinterface/probe.py @@ -527,7 +527,7 @@ def set_device_channel_indices(self, channel_indices: np.ndarray | list): """ channel_indices = np.asarray(channel_indices, dtype=int) if channel_indices.size != self.get_contact_count(): - ValueError( + raise ValueError( f"channel_indices {channel_indices.size} do not have " f"the same size as contacts {self.get_contact_count()}" ) diff --git a/tests/test_probe.py b/tests/test_probe.py index 631b6d3f..e1cd5f87 100644 --- a/tests/test_probe.py +++ b/tests/test_probe.py @@ -141,6 +141,19 @@ def test_probe(): # ~ plt.show() +def test_set_device_channel_indices_rejects_wrong_size(): + """Setting device_channel_indices with wrong count raises ValueError.""" + probe = Probe(ndim=2, si_units="um") + probe.set_contacts( + positions=np.array([[0, 0], [10, 0], [20, 0]]), + shapes="circle", + shape_params={"radius": 5}, + ) + + with pytest.raises(ValueError, match="do not have"): + probe.set_device_channel_indices([0, 1]) + + def test_probe_equality_dunder(): probe1 = generate_dummy_probe() probe2 = generate_dummy_probe()