From 3e9e238f85736a8afaae50bb54768ba61f27ca8b Mon Sep 17 00:00:00 2001 From: Ian Thompson Date: Fri, 29 Mar 2024 22:50:02 -0500 Subject: [PATCH] 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). --- mlxtend/preprocessing/tests/test_transactionencoder.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/mlxtend/preprocessing/tests/test_transactionencoder.py b/mlxtend/preprocessing/tests/test_transactionencoder.py index f5afd07e7..de5f14c5c 100644 --- 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():