exo test fix with removal of time dimension expansion for time indepe…

…ndent features
NREL · Sep 21, 2024 · cf5e466 · cf5e466
1 parent 69e8344
commit cf5e466
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/rasterizers/test_exo.py b/tests/rasterizers/test_exo.py
@@ -216,7 +216,7 @@ def test_srl_extraction_h5(s_enhance):
                 assert np.argmin(dist) == gid
 
             # make sure the mean srlation makes sense
-            test_out = hr_srl[idy, idx, 0, 0]
+            test_out = hr_srl[idy, idx, 0]
             true_out = te.source_data[iloc].mean()
             assert np.allclose(test_out, true_out)
 
@@ -272,7 +272,7 @@ def test_topo_extraction_h5(s_enhance):
                 assert np.argmin(dist) == gid
 
             # make sure the mean elevation makes sense
-            test_out = hr_elev[idy, idx, 0, 0]
+            test_out = hr_elev[idy, idx, 0]
             true_out = te.source_data[iloc].mean()
             assert np.allclose(test_out, true_out)