modified: mlxtend/preprocessing/tests/test_transactionencoder.py

- Updated failing unit test to compare output directly instead of converting to numpy arrays (which results in errors unless the dtype is set to object).
rasbt · Mar 30, 2024 · 3e9e238 · 3e9e238
1 parent 473888a
commit 3e9e238
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/mlxtend/preprocessing/tests/test_transactionencoder.py b/mlxtend/preprocessing/tests/test_transactionencoder.py
@@ -78,9 +78,7 @@ def test_fit_transform():
 def test_inverse_transform():
     oht = TransactionEncoder()
     oht.fit(dataset)
-    np.testing.assert_array_equal(
-        np.array(data_sorted), np.array(oht.inverse_transform(expect))
-    )
+    assert data_sorted == oht.inverse_transform(expect)
 
 
 def test_cloning():