diff --git a/matminer/featurizers/composition/tests/test_thermo.py b/matminer/featurizers/composition/tests/test_thermo.py index 28ba34da9..6a6b7deab 100644 --- a/matminer/featurizers/composition/tests/test_thermo.py +++ b/matminer/featurizers/composition/tests/test_thermo.py @@ -9,14 +9,23 @@ class ThermoFeaturesTest(CompositionFeaturesTest): def test_cohesive_energy(self): - mpr = MPRester() + try: + mpr = MPRester() + except ValueError: + raise SkipTest("Materials Project API key not set; Skipping cohesive energy test") + if not mpr.api_key: raise SkipTest("Materials Project API key not set; Skipping cohesive energy test") + df_cohesive_energy = CohesiveEnergy().featurize_dataframe(self.df, col_id="composition") self.assertAlmostEqual(df_cohesive_energy["cohesive energy"][0], 5.179358342, 2) def test_cohesive_energy_mp(self): - mpr = MPRester() + try: + mpr = MPRester() + except ValueError: + raise SkipTest("Materials Project API key not set; Skipping cohesive energy test") + if not mpr.api_key: raise SkipTest("Materials Project API key not set; Skipping cohesive energy test") ce = CohesiveEnergyMP()