1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
import plotly.graph_objects as go
import pytest
from boinor.plotting.orbit.backends._base import OrbitPlotterBackend
from boinor.plotting.orbit.backends.plotly import (
BasePlotly,
Plotly2D,
Plotly3D,
)
class Dummy2D(BasePlotly):
def __init__(self, figure):
theme = "plotly_dark"
# Declare the layout and attach it to the figure
layout = go.Layout(
autosize=True,
xaxis=dict(constrain="domain"),
yaxis=dict(scaleanchor="x"),
template=theme,
)
super().__init__(figure, layout)
def test_orbit_backends_plotly():
bp = Dummy2D(None)
with pytest.raises(
NotImplementedError,
match="This method is expected to be overridden by a plotting backend class.",
):
_ = bp.draw_impulse(
1, color=2, label=3, size=4
) # just dummy parameter and not needed in this function
x = (10.0, 11.1, 12.2)
y = (20.1, 21.2, 22.3)
z = (30.1, 31.2, 32.3)
p2d = Plotly2D()
v = p2d.draw_marker([x, y], color=2, label=None, marker_symbol="x", size=5)
assert v.x == x
assert v.y == y
p3d = Plotly3D()
v = p3d.draw_marker(
[x, y, z], color=2, label=None, marker_symbol="x", size=5
)
assert v.x == x
assert v.y == y
assert v.z == z
def test_orbit_backends_base():
dummyScene = 1
wrongName = "nothing"
with pytest.raises(
ValueError, match="Backend name must end with '2D' or '3D'."
):
opb = OrbitPlotterBackend(dummyScene, wrongName)
goodName = "Matplotlib2D"
opb = OrbitPlotterBackend(dummyScene, goodName)
assert opb.is_3D is False
with pytest.raises(
NotImplementedError,
match="This method is expected to be overridden by a plotting backend class.",
):
_ = opb._get_colors(
"0x000000", False
) # just dummy parameter and not needed in this function
with pytest.raises(
NotImplementedError,
match="This method is expected to be overridden by a plotting backend class.",
):
_ = opb.draw_marker(
1, color=2, label=3, marker_symbol=4, size=5
) # just dummy parameter and not needed in this function
with pytest.raises(
NotImplementedError,
match="This method is expected to be overridden by a plotting backend class.",
):
_ = opb.draw_position(
1, color=2, label=3, size=4
) # just dummy parameter and not needed in this function
with pytest.raises(
NotImplementedError,
match="This method is expected to be overridden by a plotting backend class.",
):
_ = opb.draw_impulse(
1, color=2, label=3, size=4
) # just dummy parameter and not needed in this function
with pytest.raises(
NotImplementedError,
match="This method is expected to be overridden by a plotting backend class.",
):
_ = opb.draw_sphere(
1, color=2, label=3, radius=4
) # just dummy parameter and not needed in this function
with pytest.raises(
NotImplementedError,
match="This method is expected to be overridden by a plotting backend class.",
):
_ = opb.undraw_attractor()
with pytest.raises(
NotImplementedError,
match="This method is expected to be overridden by a specific plotting backend.",
):
_ = opb.draw_coordinates(
1, colors=2, label=3, size=4
) # just dummy parameter and not needed in this function
with pytest.raises(
NotImplementedError,
match="This method is expected to be overridden by a specific plotting backend.",
):
_ = opb.draw_axes_labels_with_length_scale_units(
1
) # just dummy parameter and not needed in this function
with pytest.raises(
NotImplementedError,
match="This method is expected to be overridden by a specific plotting backend.",
):
_ = opb.update_legend()
with pytest.raises(
NotImplementedError,
match="This method is expected to be overridden by a specific plotting backend.",
):
_ = opb.resize_limits()
with pytest.raises(
NotImplementedError,
match="This method is expected to be overridden by a specific plotting backend.",
):
_ = opb.generate_labels(
1, 2, 3
) # just dummy parameter and not needed in this function
with pytest.raises(
NotImplementedError,
match="This method is expected to be overridden by a specific plotting backend.",
):
_ = opb.show()
|
