From 16b31ec29333417038ba544e64047f2837bec5fd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana?= Date: Thu, 26 Jan 2023 11:17:10 +0100 Subject: [PATCH] test: :white_check_mark: Complete join_transform test --- src/fimdlp/tests/FImdlp_test.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/fimdlp/tests/FImdlp_test.py b/src/fimdlp/tests/FImdlp_test.py index ec7096d..315c8b8 100644 --- a/src/fimdlp/tests/FImdlp_test.py +++ b/src/fimdlp/tests/FImdlp_test.py @@ -199,6 +199,8 @@ class FImdlpTest(unittest.TestCase): computed = clf.join_transform(x, y, 0) for computed, expected in zip(computed, expected): self.assertListEqual(expected, computed.tolist()) + expected_y = [b"f00", b"f00", b"f21", b"f32", b"f43"] + self.assertListEqual(expected_y, clf.y_join_) def test_join_transform_error(self): y = ["f0", "f0", "f2", "f3", "f4"]