From 132a2932cd95dc6bda2bbe53a97af599da33a47e Mon Sep 17 00:00:00 2001
From: florian <f.spreckelsen@inidscale.com>
Date: Mon, 10 Jul 2023 12:56:11 +0200
Subject: [PATCH] TST: Add unittest for empty arrays
---
unittests/test_table_importer.py | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/unittests/test_table_importer.py b/unittests/test_table_importer.py
index 0b3f0d7c..dd5b7af7 100644
--- a/unittests/test_table_importer.py
+++ b/unittests/test_table_importer.py
@@ -181,6 +181,15 @@ class TableImporterTest(unittest.TestCase):
self.assertEqual(df_new.shape[1], 4)
self.assertEqual(df_new.iloc[0].b, 5)
+ # check that missing array-valued fields are detected correctly:
+ df = pd.DataFrame([[[None, None], 4, 2.0, 'yes'],
+ ['b', 5, 3.0, 'no']],
+ columns=['a', 'b', 'c', 'd'])
+ df_new = importer.check_missing(df)
+ self.assertEqual(df_new.shape[0], 1)
+ self.assertEqual(df_new.shape[1], 4)
+ self.assertEqual(df_new.iloc[0].b, 5)
+
def test_wrong_datatype(self):
importer = TableImporter(**self.importer_kwargs)
df = pd.DataFrame([[None, np.nan, 2.0, 'yes'],
--
GitLab