def test_comp_descriptor_1():
desc = Compositions(n_jobs=1)
desc.fit_transform(pd.Series([{'H': 2}], name='composition'))
desc.fit_transform(pd.Series([{'H': 2}], name='other'))
tmp1 = desc.fit_transform(pd.Series([{
'H': 2
}], name='other'),
composition='other')
tmp2 = desc.fit_transform([{'H': 2}])
assert tmp1.shape == (1, 290)
assert isinstance(tmp1, pd.DataFrame)
assert isinstance(tmp2, pd.DataFrame)
assert np.all(tmp1.values == tmp2.values)
tmp = desc.transform([{'H': 2}], featurizers=['WeightedAverage'])
assert tmp.shape == (1, 58)
tmp = desc.transform([{'H': 2}], featurizers='all')
assert tmp.shape == (1, 500)
tmp = desc.transform([{'H': 2}], featurizers=Compositions.classic)
assert tmp.shape == (1, 290)
def test_comp_descriptor_1():
desc = Compositions(n_jobs=1)
desc.fit_transform(pd.Series([{'H': 2}], name='composition'))
desc.fit_transform(pd.Series([{'H': 2}], name='other'))
tmp1 = desc.fit_transform(pd.Series([{'H': 2}], name='other'), composition='other')
tmp2 = desc.fit_transform([{'H': 2}])
assert tmp1.shape == (1, 290)
assert isinstance(tmp1, pd.DataFrame)
assert isinstance(tmp2, pd.DataFrame)
assert np.all(tmp1.values == tmp2.values)
def test_comp_descripotor():
desc = Compositions(n_jobs=1)
desc.fit_transform(pd.Series([{'H': 2}], name='composition'))