BUG: test for float

pysat · Jul 23, 2020 · fae4012 · fae4012
1 parent 4bbb9c4
commit fae4012
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pysatMadrigal/tests/test_utils_coords.py b/pysatMadrigal/tests/test_utils_coords.py
@@ -30,9 +30,11 @@ def test_geodetic_to_geocentric(Nvals, input, inverse, target):
                                            inverse=inverse)
 
     for i in range(len(target)):
-        assert abs(output[i] - target[i]).max() < 1.0e-6
         if Nvals > 1:
+            assert abs(output[i] - target[i]).max() < 1.0e-6
             assert output[i].shape == lat_in.shape
+        else:
+            assert abs(output[i] - target[i]) < 1.0e-6
 
 
 def test_geodetic_to_geocentric_and_back():