Example #1
0
 def runTest(self):
     """testing multi-file dataset access"""
     f = MFDataset(self.files, check=True)
     f.set_auto_maskandscale(True)  # issue570
     f.set_always_mask(False)
     assert f.history == 'created today'
     assert_array_equal(np.arange(0, nx), f.variables['x'][:])
     varin = f.variables['data']
     datin = varin[:]
     assert_array_equal(datin.mask, data.mask)
     varin.set_auto_maskandscale(False)
     data2 = data.filled()
     assert varin.long_name == 'phony data'
     assert len(varin) == nx
     assert varin.shape == (nx, ydim, zdim)
     assert varin.dimensions == ('x', 'y', 'z')
     assert_array_equal(varin[4:-4:4, 3:5, 2:8], data2[4:-4:4, 3:5, 2:8])
     assert varin[0, 0, 0] == data2[0, 0, 0]
     assert_array_equal(varin[:], data2)
     assert getattr(varin, 'nonexistantatt', None) == None
     f.close()
     # test master_file kwarg (issue #835).
     f = MFDataset(self.files, master_file=self.files[-1], check=True)
     assert_array_equal(np.arange(0, nx), f.variables['x'][:])
     varin = f.variables['data']
     assert_array_equal(varin[4:-4:4, 3:5, 2:8], data2[4:-4:4, 3:5, 2:8])
     f.close()
     # testing multi-file get_variables_by_attributes
     f = MFDataset(self.files, check=True)
     assert f.get_variables_by_attributes(axis='T') == []
     f.get_variables_by_attributes(units='zlotys')[0] == f['x']
     f.close()
Example #2
0
 def runTest(self):
     """testing multi-file dataset access"""
     f = MFDataset(self.files,check=True)
     f.set_auto_maskandscale(True) # issue570
     f.set_always_mask(False)
     assert f.history == 'created today'
     assert_array_equal(np.arange(0,nx),f.variables['x'][:])
     varin = f.variables['data']
     datin = varin[:]
     assert_array_equal(datin.mask,data.mask)
     varin.set_auto_maskandscale(False)
     data2 = data.filled()
     assert varin.long_name == 'phony data'
     assert len(varin) == nx
     assert varin.shape == (nx,ydim,zdim)
     assert varin.dimensions == ('x','y','z')
     assert_array_equal(varin[4:-4:4,3:5,2:8],data2[4:-4:4,3:5,2:8])
     assert varin[0,0,0] == data2[0,0,0]
     assert_array_equal(varin[:],data2)
     assert getattr(varin,'nonexistantatt',None) == None
     f.close()
     # test master_file kwarg (issue #835).
     f = MFDataset(self.files,master_file=self.files[-1],check=True)
     assert_array_equal(np.arange(0,nx),f.variables['x'][:])
     varin = f.variables['data']
     assert_array_equal(varin[4:-4:4,3:5,2:8],data2[4:-4:4,3:5,2:8])
     f.close()
     # testing multi-file get_variables_by_attributes
     f = MFDataset(self.files,check=True)
     assert f.get_variables_by_attributes(axis='T') == []
     f.get_variables_by_attributes(units='zlotys')[0] == f['x']
     f.close()
Example #3
0
 def test_get_by_mfdataset(self):
     """testing multi-file get_variables_by_attributes."""
     f = MFDataset(self.files, check=True)
     assert f.get_variables_by_attributes(axis='T') == []
     f.get_variables_by_attributes(units='zlotys')[0] == f['x']
     f.close()
Example #4
0
 def test_get_by_mfdataset(self):
     """testing multi-file get_variables_by_attributes."""
     f = MFDataset(self.files,check=True)
     assert f.get_variables_by_attributes(axis='T') == []
     f.get_variables_by_attributes(units='zlotys')[0] == f['x']
     f.close()