Source Code for Module dimer.archive_tests

 1   
 2  import unittest 
 3  import tempfile 
 4   
 5  import numpy as np 
 6  import pandas as pd 
 7   
 8  import archive 
 9   
10 -class TestArchive(unittest.TestCase): 
11 -    def test_basename(self): 
12          from archive import basename, archname, join, split 
13   
14          self.assertEqual(basename("/a/b.h5:ciao"), "ciao") 
15          self.assertEqual(archname("/a/b.h5:ciao"), "/a/b.h5") 
16   
17          self.assertEqual( join("/a/b.h5", "ciao"), "/a/b.h5:ciao" ) 
18   
19          self.assertEqual( ("/a/b.h5", "ciao"), split("/a/b.h5:ciao" ) ) 
20           
21 -    def test_io(self): 
22          from archive import __SPEC_SEP__, __HDF_SUFFIX__, split 
23          with tempfile.NamedTemporaryFile(suffix="."+__HDF_SUFFIX__) as fd: 
24              p,k = split( __SPEC_SEP__.join( (fd.name, "empty") ) ) 
25   
26              obj = pd.Series(np.ones( (5,) ), index=range(5, 10) ) 
27               
28              archive.save_object(p, k, obj) 
29              ro = archive.load_object(p, k) 
30              self.assertTrue( np.all(ro.values == obj.values) ) 
31   
32 -class TestExp(unittest.TestCase): 
33 -    def test_trname(self): 
34          import experiment 
35           
36          tn = experiment.this_train_name("/a/something.cfg") 
37          print tn 
38          self.assertTrue( tn.startswith("train_something_") ) 
39