|
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