def test__write_gridext_file():
known_filename = 'pygridtools/tests/baseline_files/testgridext.inp'
result_filename = 'pygridtools/tests/result_files/testgridext.inp'
df = pandas.DataFrame(np.array([
[1.25, 3, 4, 3.75],
[1.75, 4, 4, 3.25],
[1.25, 4, 5, 3.75],
[1.75, 5, 5, 3.25],
]), columns=['x', 'ii', 'jj', 'y'])
iotools._write_gridext_file(df, result_filename, icol='ii', jcol='jj',
xcol='x', ycol='y')
testing.compareTextFiles(result_filename, known_filename)
def test_writeGEFDCGridextFiles(self):
known_filename = 'pygridtools/tests/baseline_files/modelgrid_gridext.inp'
result_path = 'pygridtools/tests/result_files'
result_file = 'modelgrid_gridext.inp'
self.mg.writeGEFDCGridextFile(
outputdir=result_path,
filename=result_file,
)
testing.compareTextFiles(
os.path.join(result_path, result_file),
known_filename
)
def test_dumpGridFile():
grid = testing.makeSimpleGrid()
outputfile = 'pygridtools/tests/result_files/grid.out'
iotools.dumpGridFiles(grid, outputfile)
if sys.platform == 'win32':
baselinefile = 'pygridtools/tests/baseline_files/grid_win.out'
elif sys.platform == 'darwin':
baselinefile = 'pygridtools/tests/baseline_files/grid_mac.out'
else:
baselinefile = 'pygridtools/tests/baseline_files/grid.out'
testing.compareTextFiles(outputfile, baselinefile)
def test_writeGEFDCControlFile(self):
known_filename = 'pygridtools/tests/baseline_files/modelgrid_gefdc.inp'
result_path = 'pygridtools/tests/result_files'
result_file = 'modelgrid_gefdc.inp'
self.mg.writeGEFDCControlFile(
outputdir=result_path,
filename=result_file,
title='Model Grid Test'
)
testing.compareTextFiles(
os.path.join(result_path, result_file),
known_filename
)
def test_baseline(self):
iotools._write_gridout_file(self.x, self.y, self.result_filename)
testing.compareTextFiles(self.result_filename, self.known_filename)
def test__write_gefdc_control_file():
known_filename = 'pygridtools/tests/baseline_files/maingefdc.inp'
result_filename = 'pygridtools/tests/result_files/maingefdc.inp'
iotools._write_gefdc_control_file(result_filename, 'Test Input File', 100, 25, 0)
testing.compareTextFiles(result_filename, known_filename)
def test_chunked(self):
iotools._write_cellinp(self.cells, self.chunked_output, maxcols=5)
testing.compareTextFiles(
self.chunked_output,
self.known_chunked_output
)
def test_basic(self):
iotools._write_cellinp(self.cells, self.basic_output)
testing.compareTextFiles(
self.basic_output,
self.known_basic_output
)
