|
1 | 1 | from probeinterface import Probe, ProbeGroup |
2 | 2 | from probeinterface import generate_dummy_probe, generate_dummy_probe_group |
3 | | -from probeinterface.plotting import plot_probe, plot_probe_group |
| 3 | +from probeinterface.plotting import plot_probe, plot_probegroup |
4 | 4 |
|
5 | 5 | import matplotlib.pyplot as plt |
6 | 6 | import numpy as np |
@@ -29,18 +29,18 @@ def test_plot_probe(): |
29 | 29 | plot_probe(probe, show_channel_on_click=True) |
30 | 30 |
|
31 | 31 |
|
32 | | -def test_plot_probe_group(): |
| 32 | +def test_plot_probegroup(): |
33 | 33 | probegroup = generate_dummy_probe_group() |
34 | 34 |
|
35 | | - plot_probe_group(probegroup, same_axes=True, with_contact_id=True) |
36 | | - plot_probe_group(probegroup, same_axes=False) |
| 35 | + plot_probegroup(probegroup, same_axes=True, with_contact_id=True) |
| 36 | + plot_probegroup(probegroup, same_axes=False) |
37 | 37 |
|
38 | 38 | # 3d |
39 | 39 | probegroup_3d = ProbeGroup() |
40 | 40 | for probe in probegroup.probes: |
41 | 41 | probegroup_3d.add_probe(probe.to_3d()) |
42 | 42 | probegroup_3d.probes[-1].move([0, 150, -50]) |
43 | | - plot_probe_group(probegroup_3d, same_axes=True) |
| 43 | + plot_probegroup(probegroup_3d, same_axes=True) |
44 | 44 |
|
45 | 45 |
|
46 | 46 | if __name__ == "__main__": |
|
0 commit comments