Skip to content

Commit 7bffeb6

Browse files
committed
Failing test
1 parent b44af37 commit 7bffeb6

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

scripts/tests/test_csv.py

+7-5
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,10 @@
1919
print("\nLoaded DataFrame:")
2020
print(df_loaded)
2121

22-
# Check if numpy arrays were converted to strings
23-
if isinstance(df_loaded["embedding"][0], str):
24-
print("\nTest Passed: NumPy arrays were converted to strings in CSV")
25-
else:
26-
print("\nTest Failed: NumPy arrays were not converted as expected")
22+
# ✅ **Make the test fail by checking if we correctly load NumPy arrays**
23+
try:
24+
assert isinstance(df_loaded["embedding"][0], np.ndarray), "Test Failed: Embeddings were not preserved as NumPy arrays!"
25+
print("\nTest Passed: Embeddings were correctly preserved as NumPy arrays")
26+
except AssertionError as e:
27+
print("\nTest Failed: Pandas does not preserve NumPy arrays in CSV, needs improvement!")
28+
raise e

0 commit comments

Comments
 (0)