use None in test

cwhanse · Apr 9, 2024 · fc2fd20 · fc2fd20
1 parent e4e2219
commit fc2fd20
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pvanalytics/tests/features/test_snow.py b/pvanalytics/tests/features/test_snow.py
@@ -29,7 +29,7 @@ def test_categorize():
     # np.nan, vr<thres, vr<thres, vr=thres, vr>thres, vr>thres, vr<thres
     # vo>thres, vo>thres, vo>thres, vo>thres, vo>thres, vo>thres, vo<thres
     # tr<thres, np.nan, tr>thres, tr<thres, tr<thres, tr<thres, tr>thres
-    expected = np.array([np.nan, np.nan, 2, 2, 2, 4, 0])
+    expected = np.array([None, None, 2, 2, 2, 4, 0])
     result = snow.categorize(vmp_ratio, transmission, voltage, min_dcv,
                              threshold_vratio, threshold_transmission)
     assert_array_equal(result, expected)