Example #1
0
 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, ))
Example #2
0
 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']))