def test_crystal_4(data): cg = CrystalGraphFeaturizer(atom_feature=lambda s: [1, 2, 3, 4], n_jobs=1) tmp = cg.transform(data) assert isinstance(tmp, pd.DataFrame) assert tmp.shape == (2, 2) assert tmp.values[0, 0][0].numpy().tolist() == [1, 2, 3, 4]
def test_crystal_3(data): cg = CrystalGraphFeaturizer() tmp = cg.transform(data) assert isinstance(tmp, pd.DataFrame) assert tmp.shape == (2, 2) assert tmp.values[0, 0].shape == (16, 92) assert tmp.values[0, 1][0].shape == (16, 12, 41) assert tmp.values[0, 1][1].shape == (16, 12)