def test_fit_transform(self): a = Autocorrelation(depths=[1], properties=['I']) expected = numpy.array([ [8], [8], [104], ]) self.assertTrue((a.fit_transform(ALL_DATA) == expected).all())
def test_get_labels(self): a = Autocorrelation(depths=[1, 2], properties=['I', 'EN']) X = a.fit_transform([METHANE]) labels = a.get_labels() self.assertEqual(X.shape[1], len(labels)) expected = ( 'EN_1', 'EN_2', 'I_1', 'I_2', ) self.assertEqual(labels, expected)