def test_AtomicFitPredictMixin__predict(self): X = self.X y = self.y db = DBSCAN() db.__class__ = type('newClass', (type(db), AtomicFitPredictMixin), {}) db.fit(X=X, y=y) result_db = db.predict_dict(X=X) self.assertEqual(list(result_db.keys()), ['predict']) self.assertEqual(result_db['predict'].shape, (self.size, ))
def test_AtomicFitPredictMixin__get_predict_signature(self): db = DBSCAN() db.__class__ = type('newClass', (type(db), AtomicFitPredictMixin), {}) result_db = db._get_predict_signature() self.assertEqual(sorted(result_db), sorted(['X', 'y', 'sample_weight']))