fixing test

SnowEx · Sep 5, 2023 · b39ab67 · b39ab67
1 parent 5fd5fef
commit b39ab67
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_wetsnow.py b/tests/test_wetsnow.py
@@ -503,6 +503,7 @@ def test_perma_wet_seasons(self):
         alt_wet_flag = np.full((10, 10, n), 0.0)
         freeze_flag = np.full((10, 10, n), 0.0)
         s1 = np.random.randn(10, 10, n, 3)
+        ims = np.full((10, 10, n), 4, dtype = int)
 
         x = np.linspace(0, 9, 10)
         y = np.linspace(10, 19, 10)
@@ -515,7 +516,8 @@ def test_perma_wet_seasons(self):
                         wet_flag = (["x", "y", "time"], wet_flag),
                         alt_wet_flag = (["x", "y", "time"], alt_wet_flag),
                         freeze_flag = (["x", "y", "time"], freeze_flag),
-                        s1 = (["x", "y", "time", "band"], s1)
+                        s1 = (["x", "y", "time", "band"], s1),
+                        ims = (["x", "y", "time"], ims)
                     ),
             coords = dict(
                         lon = (["x", "y"], lon),