def test_get_file_unpickling_error(self): desc = "desc" aid = "fakepkl" r = Store("./tests/test_paths/") output = io.StringIO() sys.stdout = output r.get(desc, aid, arg1="1") assert "UnpicklingError when loading file desc__fakepkl___arg1_1.pkl, consider deleting result and recomputing\n" == output.getvalue( )
def test_get_with_list(self): r3 = Store("./tests/test_paths") res = r3.get("soap", ['455'], rcut=5.0, nmax=9, lmax=9) assert type(res) == dict assert type(res['455']) is not type(None) assert len(res) == 1
def test_get(self): r3 = Store("./tests/test_paths") res = r3.get("soap", '455', rcut=5.0, nmax=9, lmax=9) assert type(res) is np.ndarray