Update test_windspharm.py

shenyulu · Dec 10, 2023 · a5b8ded · a5b8ded
1 parent 7a17b46
commit a5b8ded
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/test/test_windspharm.py b/test/test_windspharm.py
@@ -114,22 +114,22 @@ def test_calc_helmholtz():
         u_data = u_data_sample,
         v_data = v_data_sample,
     )
-    result_data1 = result_data['uchi'].sel(lon = slice(lon_start, lon_end), lat = slice(lat_end, lat_start)).data.flatten()
+    result_data1 = result_data['uchi'].sel(lon = slice(lon_start, lon_end), lat = slice(lat_end, lat_start)).data.flatten()[0:6]
     result_data1 = round_sf_np_new(result_data1)
-    result_data2 = result_data['vchi'].sel(lon = slice(lon_start, lon_end), lat = slice(lat_end, lat_start)).data.flatten()
+    result_data2 = result_data['vchi'].sel(lon = slice(lon_start, lon_end), lat = slice(lat_end, lat_start)).data.flatten()[0:6]
     result_data2 = round_sf_np_new(result_data2)
-    result_data3 = result_data['upsi'].sel(lon = slice(lon_start, lon_end), lat = slice(lat_end, lat_start)).data.flatten()
+    result_data3 = result_data['upsi'].sel(lon = slice(lon_start, lon_end), lat = slice(lat_end, lat_start)).data.flatten()[0:6]
     result_data3 = round_sf_np_new(result_data3)
-    result_data4 = result_data['vpsi'].sel(lon = slice(lon_start, lon_end), lat = slice(lat_end, lat_start)).data.flatten()
+    result_data4 = result_data['vpsi'].sel(lon = slice(lon_start, lon_end), lat = slice(lat_end, lat_start)).data.flatten()[0:6]
     result_data4 = round_sf_np_new(result_data4)
     refer_data = xr.open_dataset(os.path.join(TEST_DATA_PATH, 'test_output_calc_helmholtz.nc'))
-    refer_data1 = refer_data['uchi'].sel(lon = slice(lon_start, lon_end), lat = slice(lat_end, lat_start)).data.flatten()
+    refer_data1 = refer_data['uchi'].sel(lon = slice(lon_start, lon_end), lat = slice(lat_end, lat_start)).data.flatten()[0:6]
     refer_data1 = round_sf_np_new(refer_data1)
-    refer_data2 = refer_data['vchi'].sel(lon = slice(lon_start, lon_end), lat = slice(lat_end, lat_start)).data.flatten()
+    refer_data2 = refer_data['vchi'].sel(lon = slice(lon_start, lon_end), lat = slice(lat_end, lat_start)).data.flatten()[0:6]
     refer_data2 = round_sf_np_new(refer_data2)
-    refer_data3 = refer_data['upsi'].sel(lon = slice(lon_start, lon_end), lat = slice(lat_end, lat_start)).data.flatten()
+    refer_data3 = refer_data['upsi'].sel(lon = slice(lon_start, lon_end), lat = slice(lat_end, lat_start)).data.flatten()[0:6]
     refer_data3 = round_sf_np_new(refer_data3)
-    refer_data4 = refer_data['vpsi'].sel(lon = slice(lon_start, lon_end), lat = slice(lat_end, lat_start)).data .flatten()
+    refer_data4 = refer_data['vpsi'].sel(lon = slice(lon_start, lon_end), lat = slice(lat_end, lat_start)).data .flatten()[0:6]
     refer_data4 = round_sf_np_new(refer_data4)
     assert np.isclose(result_data1, refer_data1).all()
     assert np.isclose(result_data2, refer_data2).all()