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)