Updated test of event_end_units

cta-observatory · Nov 23, 2023 · 22a6a05 · 22a6a05
1 parent 1515a86
commit 22a6a05
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/test_units.py b/tests/test_units.py
@@ -25,8 +25,10 @@ def test_new_field():
 
 
 def test_event_end_units():
-
-    assert all([f.unit is None for f in event_end_fields])
+
+    for version in event_end_fields:
+
+        assert all([f.unit is None for f in event_end_fields[version]])
 
 
 def test_run_end_units():