def test_dumpSWMM5Format_KeepZeros(self):
data = exporters.SWMM5Format(
self.fivemin,
'Test-Station',
col='precip',
freq='5min',
dropzeros=False,
filename=getTestFile('test_dumpSWMM_withZeros.dat'))
ntools.assert_greater(data[data.precip == 0].shape[0], 0)
def test_dumpSWMM5Format_form(self):
data = exporters.SWMM5Format(self.fivemin,
'Test-Station',
col='precip',
freq='5min',
dropzeros=True,
filename=getTestFile('test_dumpSWMM.dat'))
ntools.assert_true(isinstance(data, pandas.DataFrame))
ntools.assert_list_equal(data.columns.tolist(), self.known_columns)
def test_dumpSWMM5Format_ResultHourly(self):
testfilename = getTestFile('test_dumpSWMM_hourly.dat')
data = exporters.SWMM5Format(self.fivemin,
'Test-Station',
col='precip',
freq='hourly',
dropzeros=False,
filename=testfilename)
with open(self.known_hourly_swmm5_file, 'r') as f:
known_data = f.read()
with open(testfilename, 'r') as f:
test_data = f.read()
ntools.assert_equal(known_data, test_data)
