add one more check in tests

xsuite · Nov 20, 2024 · 1f9a0d4 · 1f9a0d4
1 parent 71ca75b
commit 1f9a0d4
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests/test_utilities.py b/tests/test_utilities.py
@@ -596,13 +596,17 @@ def test_element_from_wake_and_impedance_table():
 
         assert component_found
 
+        assert hasattr(component, 'wake')
+
         if hasattr(component, 'wake'):
             test_wake = np.interp(test_points_wake, wake_table['time'],
                                   wake_table[component_kind])
 
             xo.assert_allclose(component.wake(test_points_wake), test_wake,
                                rtol=1e-3)
 
+        hasattr(component, 'impedance')
+
         if hasattr(component, 'impedance'):
             test_impedance = np.interp(test_points_impedance,
                                        impedance_table['frequency'],