def test_all_interstice_and_packing_mro(self):
all_df = self.cluster_packing_df.join(self.atomic_packing_df)
all_df = all_df.join(self.sro_volume_area_df)
mro = MRO(save=False)
result_df = mro.fit_transform(X=all_df, dependent_df=self.nn_df)
self.assertEqual(len(result_df.columns), 72)
self.assertEqual(result_df.iloc[3, -1], -0.026077608080648407)
def test_voro_index_mro(self):
mro = MRO(save=False)
result_df = mro.fit_transform(X=self.voro_idx_df,
dependent_df=self.nn_df)
self.assertTrue('MRO_mean_Voronoi_idx_5_voro' in result_df.columns)
self.assertEqual(len(result_df), len(self.voro_idx_df))
self.assertAlmostEqual(
result_df.iloc[4]['MRO_mean_Voronoi_idx_3_voro'], 0.777778, 6)
self.assertAlmostEqual(result_df.iloc[4]['MRO_std_Voronoi_idx_3_voro'],
1.227262, 6)
self.assertAlmostEqual(result_df.iloc[4]['MRO_min_Voronoi_idx_3_voro'],
0.000000, 6)
self.assertAlmostEqual(result_df.iloc[4]['MRO_max_Voronoi_idx_3_voro'],
4.000000, 6)
self.assertAlmostEqual(result_df.iloc[4]['MRO_Avg_5_fold_symm_idx'],
0.375000, 6)
self.assertAlmostEqual(result_df.iloc[4]['MRO_Avg_6_fold_symm_idx'],
0.250000, 6)
def test_cluster_packing_mro(self):
mro = MRO(save=False, stat_ops=['sum', 'mean', 'std', 'min', 'max'])
result_df = mro.fit_transform(X=self.cluster_packing_df,
dependent_df=self.nn_df)
self.assertEqual(len(result_df.columns), 5)
self.assertEqual(result_df.iloc[7, 0], 11.287025655788824)
def test_atomic_packing_mro(self):
mro = MRO(save=False, stats_types=[0, 1, 1, 1, 1, 0])
result_df = mro.fit_transform(X=self.atomic_packing_df,
dependent_df=self.nn_df)
self.assertEqual(len(result_df.columns), 4)
self.assertEqual(result_df.iloc[5, 0], 0.18774192756735897)
def test_volume_area_interstice_mro(self):
mro = MRO(save=False, stats_types="all")
result_df = mro.fit_transform(X=self.sro_volume_area_df,
dependent_df=self.nn_df)
self.assertEqual(len(result_df.columns), 60)
self.assertEqual(result_df.iloc[3, -1], -0.026077608080648407)