def test_get_file_missing_param(self):
desc = "soap"
aid = "aid_111"
r4 = Store("./tests/results/")
res = np.array([[1, 2, 3], [4, 5, 6]])
r4.store(res, desc, aid, rcut=9, nmax=10, lmax=10)
ret_val2 = r4._get_file(aid, desc, rcut=9, lmax=10)
assert ret_val2 == None
shutil.rmtree("./tests/results/")
def test_get_file_string(self):
r3 = Store("./tests/results")
result = "Random test result"
desc = "test"
aid = "12"
r3.store(result, desc, aid, arg_a=1, arg_b=2)
ret_val3 = r3._get_file(desc, aid, arg_a=1, arg_b=2)
assert ret_val3 == result
shutil.rmtree("./tests/results/")
def test_get_file_numpy_array(self):
'''Tests to make sure get_descriptor returns expected value'''
desc = "soap"
aid = "aid_111"
r4 = Store("./tests/results/")
res = np.array([[1, 2, 3], [4, 5, 6]])
r4.store(res, desc, aid, rcut=9, nmax=10, lmax=10)
ret_val = r4._get_file(desc, aid, rcut=9, nmax=10, lmax=10)
assert np.array_equal(ret_val, res)
shutil.rmtree("./tests/results/")