def test_manual_edges_1d(self): man_g = Grid(np.array([1,2,3]), edges=[1.0,-1.0,-3.0,-5.0]) self.assertTrue( man_g.get_edges() == [[1.0,-1.0,-3.0,-5.0]] )
def test_manual_edges_3d(self): man_edges = [range(6),range(6,0,-1),range(0,-6,-1)] man_edges = [ list(e) for e in man_edges ] man_g = Grid(np.array(self.three_d), edges=man_edges) self.assertTrue( man_g.get_edges() == man_edges )
def test_default_edges_1d(self): edges = [[i-.5 for i in range( len(self.g)+1)]] grd = Grid( self.one_d.copy() ) self.assertTrue( grd.get_edges() == edges )
def test_default_edges_3d(self): grd = Grid( self.three_d.copy() ) edges = [[i-.5 for i in range(e+1)] for e in self.g.shape] self.assertTrue( grd.get_edges() == edges )