From 5a55c97c3139f28a2e5e914efa3c3d8a0d7467e8 Mon Sep 17 00:00:00 2001 From: Simon Liu Date: Mon, 6 Nov 2023 14:37:25 -0800 Subject: [PATCH] update test to test for nan --- add_collection_test.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/add_collection_test.py b/add_collection_test.py index d0a029bf..a3bc1db4 100644 --- a/add_collection_test.py +++ b/add_collection_test.py @@ -136,11 +136,15 @@ def verify_variables(merged_group, origin_group, subset_index, both_merged): merged_data = np.resize(merged_var[subset_index], origin_var.shape) origin_data = origin_var + equal_nan = True + if merged_data.dtype.kind == 'S': + equal_nan = False + # verify variable data if isinstance(origin_data, str): unittest.TestCase().assertEqual(merged_data, origin_data) else: - unittest.TestCase().assertTrue(np.array_equal(merged_data, origin_data, equal_nan=True)) + unittest.TestCase().assertTrue(np.array_equal(merged_data, origin_data, equal_nan=equal_nan)) def verify_groups(merged_group, origin_group, subset_index, file=None, both_merged=False): @@ -221,6 +225,7 @@ def test(collection_id, venue): print('\nDone downloading.') filename = file_names[0] + # Handle time dimension and variables dropping merge_dataset = nc.Dataset(filename, 'r')