class CitrineDataRetrievalTest(unittest.TestCase):
def setUp(self):
self.cdr = CitrineDataRetrieval(citrine_key)
def test_get_data(self):
pifs_lst = self.cdr.get_api_data(formula="W",
data_type='EXPERIMENTAL',
max_results=10)
self.assertEqual(len(pifs_lst), 10)
df = self.cdr.get_dataframe(criteria={
'formula': 'W',
'data_type': 'EXPERIMENTAL',
'max_results': 10
},
print_properties_options=False)
self.assertEqual(df.shape[0], 10)
def test_multiple_items_in_list(self):
df = self.cdr.get_dataframe(criteria={
'data_set_id': 114192,
'max_results': 102
},
print_properties_options=False)
self.assertEqual(df.shape[0], 102)
test_cols = {
"Thermal conductivity_5-conditions", "Condition_1",
"Thermal conductivity_10"
}
self.assertTrue(test_cols < set(df.columns))
class CitrineDataRetrievalTest(unittest.TestCase):
def setUp(self):
self.cdr = CitrineDataRetrieval(citrine_key)
def test_get_data(self):
pifs_lst = self.cdr.get_api_data(formula="W",
data_type='EXPERIMENTAL',
max_results=10)
df = self.cdr.get_dataframe(pifs_lst)
assert df.shape[0] == 10
def test_mutiple_items_in_list(self):
pifs_lst = self.cdr.get_api_data(data_set_id=114192, max_results=102)
df = self.cdr.get_dataframe(pifs_lst)
assert df.shape[0] == 102
for col in [
"Thermal conductivity_5-conditions", "Condition_1",
"Thermal conductivity_10"
]:
assert col in df.columns
