Add smoke test for sklearnex (Kaggle#1143)

sklearnex was added in Kaggle#1135 http://227237481
qingtaozhou · Mar 29, 2022 · 98ca09f · 98ca09f
1 parent f4d7262
commit 98ca09f
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/test_sklearnex.py b/tests/test_sklearnex.py
@@ -0,0 +1,12 @@
+import unittest
+
+import numpy as np
+
+class TestSklearnex(unittest.TestCase):
+    def test_dbscan(self):
+        from sklearnex.cluster import DBSCAN
+        X = np.array([[1., 2.], [2., 2.], [2., 3.],
+            [8., 7.], [8., 8.], [25., 80.]], dtype=np.float32)
+
+        clustering = DBSCAN(eps=3, min_samples=2).fit(X)
+        np.testing.assert_array_equal(np.array([0, 0, 0, 1, 1, -1]), clustering.labels_)