Skip to content

Commit 9b32181

Browse files
committed
TST: created test for get_instante_attributes() with flight without rail buttons
1 parent f9f57a4 commit 9b32181

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

tests/unit/test_utilities.py

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,7 @@ def test_flutter_plots(mock_show, flight_calisto_custom_wind): # pylint: disabl
165165
), "An error occurred while running the utilities._flutter_plots function."
166166

167167

168-
def test_get_instance_attributes(flight_calisto_robust):
168+
def test_get_instance_attributes_with_robust_flight(flight_calisto_robust):
169169
"""Tests if get_instance_attributes returns the expected results for a
170170
robust flight object."""
171171

@@ -178,6 +178,19 @@ def test_get_instance_attributes(flight_calisto_robust):
178178
assert attr == value
179179

180180

181+
def test_get_instance_attributes_with_flight_without_rail_buttons(flight_calisto):
182+
"""Tests if get_instance_attributes returns the expected results for a
183+
flight object that contains a rocket object without rail buttons."""
184+
185+
attributes = utilities.get_instance_attributes(flight_calisto)
186+
for key, value in attributes.items():
187+
attr = getattr(flight_calisto, key)
188+
if isinstance(attr, np.ndarray):
189+
assert np.allclose(attr, value)
190+
else:
191+
assert attr == value
192+
193+
181194
@pytest.mark.parametrize(
182195
"f, eps, expected",
183196
[

0 commit comments

Comments
 (0)