def maggma_stores():
return {
"sgroups":
JSONStore(f"{dir_path}/maggma_sgroup_store.json", key="group_id"),
"materials":
JSONStore(f"{dir_path}/maggma_materials_store.json",
key="material_id"),
}
def setUp(self):
self.materials = JSONStore(boltztrap4dos_mat)
self.materials.connect()
self.bandstructure = JSONStore(boltztrap4dos_bs)
self.bandstructure.connect()
self.dos_ref = JSONStore(boltztrap4dos_dos)
self.dos_ref.connect()
self.dos = MemoryStore("dos")
self.dos.connect()
def jsonstore(test_dir):
files = []
for f in ["a.json", "b.json"]:
files.append(test_dir / "test_set" / f)
return JSONStore(files)
def test_json_store_load(jsonstore, test_dir):
jsonstore.connect()
assert len(list(jsonstore.query())) == 20
jsonstore = JSONStore(test_dir / "test_set" / "c.json.gz")
jsonstore.connect()
assert len(list(jsonstore.query())) == 20
def __init__(self, data_file_path):
"""
Assumers that the data_file_path contains data from
a mpquery request in JSON format.
"""
self._dataset = JSONStore(data_file_path)
self._objective_partitioned_data = {}
self._objective_weights = {}
self._objective_ids = {}
self._dataset.connect()
def setUp(self):
# Set up test db, set up mpsft, etc.
self.test_tasks = JSONStore([test_tasks])
self.elasticity = MemoryStore("atomate_tasks")
self.test_tasks.connect()
self.elasticity.connect()
def setUp(self):
self.materials = JSONStore(test_mats, lu_type='isoformat')
self.bond_valence = MemoryStore("bond_valence")
def setUp(self):
self.materials = JSONStore(test_mats,lu_type='isoformat')
self.materials.connect()
self.thermo = MemoryStore("thermo")
self.thermo.connect()
def setUp(self):
self.materials = JSONStore(test_mats, lu_type='isoformat')
self.magnetism = MemoryStore("magnetism")
def setUp(self):
self.materials = JSONStore(test_mats)
self.magnetism = MemoryStore("magnetism")
def tasks_store(test_dir):
return JSONStore(test_dir / "test_si_tasks.json.gz")
def setUp(self):
# Set up test db, set up mpsft, etc.
self.test_tasks = JSONStore(test_tasks)
self.task_types = MemoryStore("task_types")
self.test_tasks.connect()
self.task_types.connect()
def setUp(self):
self.materials = JSONStore(test_mats)
self.bond_valence = MemoryStore("bond_valence")
def setUp(self):
self.materials = JSONStore(test_mats)
self.bonding = MemoryStore("bonding")
def setUp(self):
self.materials = JSONStore(test_mats)
self.materials.connect()
self.thermo = MemoryStore("thermo")
self.thermo.connect()