Refactor numpy imports to use alias 'np'

Author: davidhassell

cf/test/test_dsg.py

@@ -5,7 +5,7 @@
 import tempfile
 import unittest
 
-import numpy
+import numpy as np
 
 faulthandler.enable()  # to debug seg faults and timeouts
 
@@ -44,7 +44,7 @@ class DSGTest(unittest.TestCase):
         "DSG_timeSeriesProfile_indexed_contiguous.nc",
     )
 
-    a = numpy.ma.masked_all((4, 9), dtype=float)
+    a = np.ma.masked_all((4, 9), dtype=float)
     a[0, 0:3] = [0.0, 1.0, 2.0]
     a[1, 0:7] = [1.0, 11.0, 21.0, 31.0, 41.0, 51.0, 61.0]
     a[2, 0:5] = [2.0, 102.0, 202.0, 302.0, 402.0]
@@ -60,7 +60,7 @@ class DSGTest(unittest.TestCase):
         8003.0,
     ]
 
-    b = numpy.array(
+    b = np.array(
         [
             [
                 [20.7, -99, -99, -99],
@@ -149,7 +149,7 @@ class DSGTest(unittest.TestCase):
         ]
     )
 
-    b = numpy.ma.where(b == -99, numpy.ma.masked, b)
+    b = np.ma.where(b == -99, np.ma.masked, b)
 
     def test_DSG_contiguous(self):
         f = cf.read(self.contiguous, verbose=0)
@@ -286,7 +286,7 @@ def test_DSG_indexed_contiguous(self):
 
     def test_DSG_create_contiguous(self):
         # Define the ragged array values
-        ragged_array = numpy.array([1, 3, 4, 3, 6], dtype="float32")
+        ragged_array = np.array([1, 3, 4, 3, 6], dtype="float32")
         # Define the count array values
         count_array = [2, 3]
 
@@ -317,7 +317,7 @@ def test_DSG_create_contiguous(self):
         self.assertTrue(
             (
                 z.data.array
-                == numpy.ma.masked_array(
+                == np.ma.masked_array(
                     data=[[1.0, 3.0, 99], [4.0, 3.0, 6.0]],
                     mask=[[False, False, True], [False, False, False]],
                     fill_value=1e20,
@@ -331,12 +331,12 @@ def test_DSG_create_contiguous(self):
         self.assertTrue(
             (
                 z.data.compressed_array
-                == numpy.array([1.0, 3.0, 4.0, 3.0, 6.0], dtype="float32")
+                == np.array([1.0, 3.0, 4.0, 3.0, 6.0], dtype="float32")
             ).all()
         )
 
         self.assertTrue(
-            (z.data.get_count().data.array == numpy.array([2, 3])).all()
+            (z.data.get_count().data.array == np.array([2, 3])).all()
         )