def testRead(self): from spacepy.pybats import SatOrbit from numpy.testing import assert_array_equal as assert_array sat = SatOrbit('data/pybats_test/testsat.dat') # Check header: self.assertEqual(sorted(sat.attrs['head']), sorted(['test', 'header', 'values'])) self.assertEqual(sat.attrs['coor'], 'SMG') # Check time and position: assert_array(sat['time'], self.time) assert_array(sat['xyz'], self.pos)
def testRead(self): from spacepy.pybats import SatOrbit from numpy.testing import assert_array_equal as assert_array sat = SatOrbit('data/pybats_test/testsat.dat') # Check header: self.assertEqual(sorted(sat.attrs['head']), sorted(['test','header','values'])) self.assertEqual(sat.attrs['coor'], 'SMG') # Check time and position: assert_array(sat['time'], self.time) assert_array(sat['xyz'], self.pos)
def testWrite(self): ''' Create a file from scratch. ''' from os import remove from spacepy.pybats import SatOrbit from numpy.testing import assert_array_equal as assert_array # New empty object: sat = SatOrbit() # Load in fake data: sat['time'] = self.time sat['xyz'] = self.pos # Add some info: sat.attrs['coor'] = 'SMG' sat.attrs['file'] = 'testsat.dat' sat.attrs['head'] = ['test', 'header', 'values'] sat.write() import os os.listdir('.') #### Now, test the integrity of the file: sat = SatOrbit('./testsat.dat') # Check header: self.assertEqual(sorted(sat.attrs['head']), sorted(['test', 'header', 'values'])) self.assertEqual(sat.attrs['coor'], 'SMG') # Check time and position: assert_array(sat['time'], self.time) assert_array(sat['xyz'], self.pos) # Get rid of file: remove('./testsat.dat')
def testWrite(self): ''' Create a file from scratch. ''' from os import remove from spacepy.pybats import SatOrbit from numpy.testing import assert_array_equal as assert_array # New empty object: sat = SatOrbit() # Load in fake data: sat['time'] = self.time sat['xyz'] = self.pos # Add some info: sat.attrs['coor'] = 'SMG' sat.attrs['file'] = 'testsat.dat' sat.attrs['head'] = ['test','header','values'] sat.write() import os os.listdir('.') #### Now, test the integrity of the file: sat = SatOrbit('./testsat.dat') # Check header: self.assertEqual(sorted(sat.attrs['head']), sorted(['test','header','values'])) self.assertEqual(sat.attrs['coor'], 'SMG') # Check time and position: assert_array(sat['time'], self.time) assert_array(sat['xyz'], self.pos) # Get rid of file: remove('./testsat.dat')
def testWrite(self): # Test that files are correctly written to file. from numpy.testing import assert_array_equal as assert_array # Save original file names: old_file_1 = self.sing.attrs['file'] old_file_2 = self.mult.attrs['file'] # Rename files, write: self.sing.attrs['file'] = './imf_sing.tmp' self.mult.attrs['file'] = './imf_mult.tmp' self.sing.write() self.mult.write() # Reopen files: sing = pb.ImfInput('./imf_sing.tmp') mult = pb.ImfInput('./imf_mult.tmp') # Ensure files were written correctly: for v in sing: assert_array(self.sing[v], sing[v]) for v in mult: assert_array(self.mult[v], mult[v])
def test_getitem(policy: P.Policy) -> None: best = policy.best_action() partial = policy[3:8].best_action() assert_array(best[3:8].numpy(), partial.numpy())