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)