Ejemplo n.º 1
0
 def test_store(self):
     '''Tests storing results as a numpy array or other type'''
     r3 = Store("./tests/results")
     result = "Random test result"
     desc = "test_desc"
     aid = "111"
     kw1 = "option_1"
     kw2 = "option_2"
     r3.store(result, desc, aid, a=kw1, b=kw2)
     fname = r3._generate_file_name(desc, aid, a=kw1, b=kw2)
     fpath = os.path.join(r3.root, desc, aid, fname)
     assert os.path.exists(fpath)
     shutil.rmtree("./tests/results/")
Ejemplo n.º 2
0
 def test_generate_file_name_diff_order(self):
     r1 = Store(".")
     f1 = r1._generate_file_name("soap", "111", nmax=11, rcut=9.0, lmax=11)
     assert f1 == "soap__111___lmax_11___nmax_11___rcut_9.0.pkl"