Fix retrieval of value

radio-astro-tools · Sep 11, 2020 · 1b7ecb3 · 1b7ecb3
1 parent 973ec90
commit 1b7ecb3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/spectral_cube/tests/test_dask.py b/spectral_cube/tests/test_dask.py
@@ -109,4 +109,8 @@ def test_statistics_consistency_casa(data_adv, tmp_path):
     pprint(stats_casa)
 
     for key in stats:
-        assert_allclose(stats[key].value, stats_casa[key])
+        if isinstance(stats[key], u.Quantity):
+            value = stats[key].value
+        else:
+            value = stats[key]
+        assert_allclose(value, stats_casa[key])