def setUp(self): ''' ThreadTest::setUp Addition to the TestBase.setUp, creates a user to use in the thread actions ''' TestBase.setUp(self) self.register(self.user)
def setUp(self): ''' ClassTest::setUp Overloaded setup call, this is response for creating a root user by default. ''' TestBase.setUp(self) self.register(self.root_user)
def setUp(self): TestBase.setUp(self) # Create a temporary directory self.test_dir = tempfile.mkdtemp() self.n = 100 self.titles = ['Array 1', 'Array 2', 'Array 3'] self.header = 'A header line' ##### Now generate output for testing #### filename = os.path.join(self.test_dir, 'test.dat') self.data = np.random.random((self.n, len(self.titles))) header = [self.header, '%d' % len(self.titles)] for ln in self.titles: header.append(ln + '\n') header = '\n'.join(header) np.savetxt(filename, self.data, delimiter=' ', header=header, comments='') # Set up the reader reader = GSLibReader() reader.AddFileName(filename) # Perform the read reader.Update() self.HEADER = reader.get_file_header() self.TABLE = reader.GetOutput()
def setUp(self): ''' QuestTest::setUp Overloaded setup call, this is responsible for creating a root user by default. Mostly useful because of the requirement of a DM user for most of the tests. ''' TestBase.setUp(self) self.register(self.root_user)
def setUp(self): TestBase.setUp(self) self.test_dir = tempfile.mkdtemp() treeMesh = """16 16 16 0.0000 0.0000 48.0000 1.000 2.000 3.000 29 1 1 1 8 9 1 1 8 1 9 1 8 9 9 1 8 1 1 9 4 5 1 9 4 9 1 9 8 1 5 9 4 5 5 9 4 1 9 9 8 9 9 9 8 1 1 13 2 3 1 13 2 5 1 13 4 1 3 13 2 3 3 13 2 1 5 13 4 5 5 13 4 1 1 15 1 2 1 15 1 3 1 15 2 1 2 15 1 2 2 15 1 1 3 15 2 3 3 15 2 1 1 16 1 2 1 16 1 1 2 16 1 2 2 16 1 """ # Write out mesh file filename = os.path.join(self.test_dir, 'octree.msh') self.meshFileName = filename with open(filename, 'w') as f: f.write(treeMesh) # write out model file(s) self.nt = 5 self.modelFileNames = ['model%d.mod' % i for i in range(self.nt)] self.modelFileNames = [ os.path.join(self.test_dir, self.modelFileNames[i]) for i in range(self.nt) ] self.arrs = [None] * self.nt for i in range(self.nt): self.arrs[i] = np.random.random(29) np.savetxt(self.modelFileNames[i], self.arrs[i], delimiter=' ', comments='! ') return
def setUp(self): ''' ForumTest::setUp Additions to the TestBase.setUp, creates a user to use in the forum actions ''' TestBase.setUp(self) self.register(self.root_user) self.logout() self.register(self.user)
def setUp(self): ''' PlayerTest::setUp Overloaded setup call, This is responsible for creating a root user by default and logging them out. Mostly useful because all of the users created subsequently will have default permissions. ''' TestBase.setUp(self) self.register(self.root_user) self.logout()
def setUp(self): TestBase.setUp(self) # Create some input tables self.t0 = vtk.vtkTable() # Populate the tables self.n = 400 self.title = 'Array 0' self.arr = np.random.random(self.n) # Table 0 self.t0.AddColumn(interface.convertArray(self.arr, self.title)) return
def setUp(self): TestBase.setUp(self) self.RTOL = 0.00001 # As higi as rotation precision can get x = np.array([0.0, 1.0, 0.0]) y = np.array([0.0, 0.0, 1.0]) z = np.array([0.0, 0.0, 0.0]) x = np.reshape(x, (len(x), -1)) y = np.reshape(y, (len(y), -1)) z = np.reshape(z, (len(z), -1)) self.pts = np.concatenate((x, y, z), axis=1) self.vtkpoints = interface.pointsToPolyData(self.pts) return
def setUp(self): TestBase.setUp(self) # create a volumetric data set self.grid = PVGeo.model_build.CreateTensorMesh().apply() # create an unudulating surface in the grid domain # make XY random in the grid bounds # make Z ranome within a very small domain inside the grid bnds = self.grid.GetBounds() x = np.random.uniform(bnds[0], bnds[1], 5000) y = np.random.uniform(bnds[2], bnds[3], 5000) z = np.random.uniform(-200, -100, 5000) self.points = interface.points_to_poly_data(np.c_[x,y,z])
def setUp(self): TestBase.setUp(self) # Create some input tables self.t0 = vtk.vtkTable() # Populate the tables self.arrs = [None, None] self.n = 400 self.titles = ('Array 0', 'Array 1') self.arrs[0] = np.random.random(self.n) # Table 0 self.arrs[1] = np.random.random(self.n) # Table 0 self.t0.AddColumn(interface.convertArray(self.arrs[0], self.titles[0])) self.t0.AddColumn(interface.convertArray(self.arrs[1], self.titles[1])) return
def setUp(self): TestBase.setUp(self) # Create some input tables self.idi = vtk.vtkImageData() self.idi.SetDimensions(20, 2, 10) self.idi.SetSpacing(1, 1, 1) self.idi.SetOrigin(100, 100, 100) # Populate the tables self.n = 400 self.title = 'Array 0' self.arr = np.random.random(self.n) self.idi.GetPointData().AddArray(interface.convert_array(self.arr, self.title)) return
def setUp(self): TestBase.setUp(self) # Create a temporary directory self.test_dir = tempfile.mkdtemp() self.mesh = """9 -300.0 -180.0 1 -130.0 1 -110.0 1 -100.0 1 100.0 40 110.0 1 130.0 1 180.0 1 300.0 1 14 -10.0 10.0 5 22.0 4 42.0 5 57.0 3 63.0 1 71.0 1 81.0 1 95.0 1 115.0 1 140.0 1 170.0 1 205.0 1 245.0 1 300.0 1 """ self.origin = (-300, 0, 10) self.nx = 48 self.nz = 27 self.shape = (self.nx, 1, self.nz) self.extent = (0, self.shape[0], 0, self.shape[1], 0, self.shape[2]) self.data_name = 'foo' ##### Now generate output for testing #### # Produce data and write out files: meshname = self._write_mesh() modname, self.data = self._write_model() # Set up the reader: reader = TensorMeshReader() reader.set_mesh_filename(meshname) # Get and test output: reader.Update() # Test the read up front for the mesh reader.add_model_file_name(modname) reader.set_data_name(self.data_name) reader.Update() # Now read the models upfront self.GRID = reader.GetOutput() return
def setUp(self): TestBase.setUp(self) # create a volumetric data set self.grid = PVGeo.model_build.CreateTensorMesh().Apply() # create a spline throught the data set def path1(y): """Equation: x = a(y-h)^2 + k""" a = -0.0001 x = a * y**2 + 1000 idxs = np.argwhere(x > 0) return x[idxs][:, 0], y[idxs][:, 0] x, y = path1(np.arange(-500.0, 1500.0, 25.0)) zo = np.linspace(9.0, 11.0, num=len(y)) coords = np.vstack((x, y, zo)).T self.points = interface.pointsToPolyData(coords)
def setUp(self): TestBase.setUp(self) # Create a temporary directory self.test_dir = tempfile.mkdtemp() self.n = 100 self.shape = (150, 200, 20) self.extent = (0, self.shape[0], 0, self.shape[1], 0, self.shape[2]) self.titles = ['Array 1', 'Array 2', 'Array 3'] ##### Now generate output for testing #### fname = os.path.join(self.test_dir, 'test.dat') self.data = np.random.random((self.n, len(self.titles))) header = ['%d %d %d' % self.shape, '%d' % len(self.titles)] for ln in self.titles: header.append(ln + '\n') header = '\n'.join(header) np.savetxt(fname, self.data, delimiter=' ', header=header, comments='') # Set up the reader self.GRID = SGeMSGridReader().Apply(fname)
def setUp(self): TestBase.setUp(self) # Create a temporary directory self.test_dir = tempfile.mkdtemp() self.filename = os.path.join(self.test_dir, 'test.dat') sample = """ncols 4 nrows 6 xllcorner 100.0 yllcorner 50.0 cellsize 50.0 NODATA_value -9999 -9999 -9999 5 2 -9999 20 100 36 3 8 35 10 32 42 50 6 88 75 27 9 13 5 1 -9999 """ with open(self.filename, 'w') as f: f.write(sample)
def setUp(self): TestBase.setUp(self) # Create a temporary directory self.test_dir = tempfile.mkdtemp() self.filename = os.path.join(self.test_dir, 'test.xyz') # Make a temporary file to test: self.nrows = 100 self.ncols = 8 # LEAVE ALONE self.header = 'X, dx, Y, dy, Z, dz, approximate distance, cell index' self.data = np.random.random((self.nrows, self.ncols)) np.savetxt(self.filename, self.data, header=self.header, comments='! ', fmt='%.6e') reader = XYZTextReader() reader.AddFileName(self.filename) reader.Update() self.TABLE = reader.GetOutput() return
def setUp(self): TestBase.setUp(self) # Create some input tables self.t0 = vtk.vtkTable() self.t1 = vtk.vtkTable() # Populate the tables self.n = 100 self.titles = ('Array 0', 'Array 1', 'Array 2') self.arrs = [None, None, None] self.arrs[0] = np.random.random(self.n) # Table 0 self.arrs[1] = np.random.random(self.n) # Table 0 self.arrs[2] = np.random.random(self.n) # Table 1 self.t0.AddColumn(interface.convertArray(self.arrs[0], self.titles[0])) self.t0.AddColumn(interface.convertArray(self.arrs[1], self.titles[1])) self.t1.AddColumn(interface.convertArray(self.arrs[2], self.titles[2])) # Now use the `CombineTables` filter: f = CombineTables() f.SetInputDataObject(0, self.t0) f.SetInputDataObject(1, self.t1) f.Update() self.TABLE = f.GetOutputDataObject(0)
def setUp(self): TestBase.setUp(self) # Create a temporary directory self.test_dir = tempfile.mkdtemp() self.commafilename = os.path.join(self.test_dir, 'comma.txt') self.tabfilename = os.path.join(self.test_dir, 'tab.txt') # Make a temporary delimited text file to test: lines = [ 'This is a header line to skip', 'int,str,float ! Comment,this line has the data array names', '5,foo,6.9', '1,bar,8.5 ! another comment', '3,oof,7.7' ] # Append newlines lines = [ln + '\n' for ln in lines] # Now write contents to files f = open(self.commafilename, 'w') f.writelines(lines) f.close() f = open(self.tabfilename, 'w') f.writelines([ln.replace(',', '\t') for ln in lines]) f.close() return
def setUp(self): TestBase.setUp(self) # Create a temporary directory self.test_dir = tempfile.mkdtemp() self.origin = (-350, -400, 0) self.xCells = '200 100 50 20*50.0 50 100 200' self.yCells = '200 100 50 21*50.0 50 100 200' self.zCells = '20*25.0 50 100 200' self.shape = (26, 27, 23) self.n = self.shape[0] * self.shape[1] * self.shape[2] self.extent = (0, self.shape[0], 0, self.shape[1], 0, self.shape[2]) self.data_name = 'foo' ##### Now generate output for testing #### # Produce data and write out files: self.meshname = self._write_mesh() self.modname, self.data = self._write_model() self.modname_multi, self.data_multi = self._write_model_multi() # Set up the reader: reader = TensorMeshReader() reader.set_mesh_filename(self.meshname) # Get and test output: reader.Update() # Read only mesh upfront reader.add_model_file_name(self.modname) reader.set_data_name(self.data_name) reader.Update() # Read models upfront self.GRID = reader.GetOutput() #### Now read mesh with multi component data # Set up the reader: reader = TensorMeshReader() reader.set_mesh_filename(self.meshname) # Get and test output: reader.Update() # Read only mesh upfront reader.add_model_file_name(self.modname_multi) reader.set_data_name(self.data_name) reader.Update() # Read models upfront self.GRID_MULTI = reader.GetOutput()
def setUp(self): TestBase.setUp(self) # create a volumetric data set self.grid = PVGeo.model_build.CreateTensorMesh().Apply()
def setUp(self): TestBase.setUp(self) self.test_dir = tempfile.mkdtemp() self.filename = os.path.join(self.test_dir, 'test.txt')
def setUp(self): TestBase.setUp(self) self.test_dir = tempfile.mkdtemp() self.filename = os.path.join(os.path.dirname(__file__), 'data/surfer-grid.grd')
def setUp(self): TestBase.setUp(self) self.tree = etree.fromstring(TestBase.TRACK_DETAILS_SAMPLE)
def setUp(self): TestBase.setUp(self) ans = self.client.createdb(name=self.dbname, conf=config) self.client.setdb(self.dbname)
def setUp(self): TestBase.setUp(self) # Create a temporary directory self.filename = os.path.join(os.path.dirname(__file__), 'data', 'vd2.asc')
def setUp(self): TestBase.setUp(self)
def setUp(self): TestBase.setUp(self) self.RTOL = 0.00001 # As high as rotation precision can get return
def setUp(self): TestBase.setUp(self) self.register()
def setUp(self): TestBase.setUp(self) self.filename = os.path.join(os.path.dirname(__file__), 'data/Craig-Chile/LdM_topo.topo')
def setUp(self): TestBase.setUp(self) self.filename = os.path.join(os.path.dirname(__file__), 'data/Craig-Chile/LdM_grav_obs.grv')
def setUp(self): TestBase.setUp(self) # Create a temporary directory self.test_dir = tempfile.mkdtemp() self.n = 100 return