Beispiel #1
0
    def test_baseadapter__get_fit_signature(self):
        lm = LinearRegression()
        gm = GaussianMixture()
        lm.__class__ = type('newClass', (type(lm), FitPredictMixin), {})
        gm.__class__ = type('newClass', (type(gm), FitPredictMixin), {})

        result_lm = lm._get_fit_signature()
        result_gm = gm._get_fit_signature()

        self.assertEqual(sorted(result_lm), sorted(['X', 'y',
                                                    'sample_weight']))
        self.assertEqual(sorted(result_gm), sorted(['X', 'y']))