Source Code for Module dimer.nnet.cspec_tests

 1   
 2   
 3  import unittest 
 4  import numpy as np 
 5  rng = np.random.RandomState() 
 6   
 7  from config_spec import DataSpec 
 8   
 9 -class Test(unittest.TestCase): 
10   
11 -    def testDataSpec(self): 
12          for i in range(5): 
13              a,b = DataSpec.batches_from_data(rng.randint(100, 1000), 5, 
14                                               rng.randint(1, 10) / 20., i, rng) 
15              self.assertEqual( list(set(a+b)), range(len(a+b)) )         
16              self.assertTrue( i in b ) 
17   
18   
19   
20  if __name__ == "__main__": 
21      #import sys;sys.argv = ['', 'Test.testName'] 
22      unittest.main() 
23