Fix weird failing test

https://app.circleci.com/pipelines/github/octoenergy/xocto/586/workflows/9b2278c5-ea72-4d9f-862d-a7be672ce21f/jobs/571 > tests/storage/test_storage.py:1229: in test_query_dataframe_with_sql > assert result_df["array_column"][0] == [1, 2, 3] > E ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()
octoenergy · Mar 28, 2024 · c0194b1 · c0194b1
1 parent 901bc09
commit c0194b1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/storage/test_storage.py b/tests/storage/test_storage.py
@@ -1226,7 +1226,11 @@ def test_query_dataframe_with_sql(self):
         assert result_df.shape == (1, 2)
 
         assert result_df["string_column"][0] == "A"
-        assert result_df["array_column"][0] == [1, 2, 3]
+        assert tuple(result_df["array_column"][0]) == (
+            1,
+            2,
+            3,
+        )
 
     def test_query_dataframe_with_sql_with_capitalised_object_in_query(self):
         dummy_df = pd.DataFrame(self.sample_dataframe)