def test_man(): """Read a management file please""" manfile = dep.read_man(get_path('man.txt')) assert manfile['nop'] == 5 assert manfile['nini'] == 2 assert manfile['nsurf'] == 2 assert manfile['nwsofe'] == 3 assert manfile['nrots'] == 1 assert manfile['nyears'] == 11 manfile = dep.read_man(get_path('man2.txt')) assert manfile['nop'] == 0
def test_man(): """Read a management file please""" manfile = dep.read_man(get_path("man.txt")) assert manfile["nop"] == 5 assert manfile["nini"] == 2 assert manfile["nsurf"] == 2 assert manfile["nwsofe"] == 3 assert manfile["nrots"] == 1 assert manfile["nyears"] == 11 manfile = dep.read_man(get_path("man2.txt")) assert manfile["nop"] == 0
def test_man(self): """Read a management file please""" manfile = dep.read_man(get_path('man.txt')) self.assertEquals(manfile['nop'], 5) self.assertEquals(manfile['nini'], 2) self.assertEquals(manfile['nsurf'], 2) self.assertEquals(manfile['nwsofe'], 3) self.assertEquals(manfile['nrots'], 1) self.assertEquals(manfile['nyears'], 11) manfile = dep.read_man(get_path('man2.txt')) self.assertEquals(manfile['nop'], 0)
def main(): """Go Main Go""" # ['id', 'CropRotationString', 'slope', # 'rainfall', 'runoff', 'detach', 'delivery']) rows = [] for root, _dirs, files in tqdm(os.walk("/i/0/ofe")): for filename in files: ofedf = read_ofe("%s/%s" % (root, filename)) # Drop any 2007 or 2018+ data ofedf = ofedf[(ofedf['date'] < datetime.date(2018, 1, 1)) & (ofedf['date'] >= datetime.date(2008, 1, 1))] # Figure out the crop string man = "%s/%s" % (root.replace( "ofe", "man"), filename.replace("ofe", "man")) try: manres = read_man(man) except Exception as exp: print("failure reading %s\n%s" % (man, exp)) continue slp = "%s/%s" % (root.replace( "ofe", "slp"), filename.replace("ofe", "slp")) slpres = read_slp(slp) soils = get_soils(slp.replace('slp', 'prj')) for ofe in ofedf['ofe'].unique(): myofe = ofedf[ofedf['ofe'] == ofe] length = slpres[ofe - 1]['x'][-1] - slpres[ofe - 1]['x'][0] slp = ((slpres[ofe - 1]['y'][0] - slpres[ofe - 1]['y'][-1]) / length) rows.append({ 'id': "%s_%s" % (filename[:-4], ofe), 'huc12': filename[:12], 'fpath': filename.split("_")[1][:-4], 'ofe': ofe, 'CropRotationString': (get_rotation_string(manres, ofe)), 'slope[1]': slp, 'soil_mukey': soils[ofe - 1], 'rainfall': -1, 'runoff[mm/yr]': myofe['runoff'].sum() / YEARS, 'detach': -1, 'length[m]': length, 'delivery[t/a/yr]': (myofe['sedleave'].sum() / YEARS / length * 4.463) }) df = pd.DataFrame(rows) df.to_csv('results.csv', index=False)
def main(): """Go Main Go""" # ['id', 'CropRotationString', 'slope', # 'rainfall', 'runoff', 'detach', 'delivery']) rows = [] for root, _dirs, files in tqdm(os.walk("/i/0/ofe")): for filename in files: ofedf = read_ofe("%s/%s" % (root, filename)) # Drop any 2007 or 2018+ data ofedf = ofedf[(ofedf['date'] < datetime.date(2018, 1, 1)) & (ofedf['date'] >= datetime.date(2008, 1, 1))] # Figure out the crop string man = "%s/%s" % (root.replace("ofe", "man"), filename.replace("ofe", "man")) try: manres = read_man(man) except Exception as exp: print("failure reading %s\n%s" % (man, exp)) continue slp = "%s/%s" % (root.replace("ofe", "slp"), filename.replace("ofe", "slp")) slpres = read_slp(slp) soils = get_soils(slp.replace('slp', 'prj')) for ofe in ofedf['ofe'].unique(): myofe = ofedf[ofedf['ofe'] == ofe] length = slpres[ofe - 1]['x'][-1] - slpres[ofe - 1]['x'][0] slp = ((slpres[ofe - 1]['y'][0] - slpres[ofe - 1]['y'][-1]) / length) rows.append({ 'id': "%s_%s" % (filename[:-4], ofe), 'huc12': filename[:12], 'fpath': filename.split("_")[1][:-4], 'ofe': ofe, 'CropRotationString': ( get_rotation_string(manres, ofe)), 'slope[1]': slp, 'soil_mukey': soils[ofe - 1], 'rainfall': -1, 'runoff[mm/yr]': myofe['runoff'].sum() / YEARS, 'detach': -1, 'length[m]': length, 'delivery[t/a/yr]': ( myofe['sedleave'].sum() / YEARS / length * 4.463) }) df = pd.DataFrame(rows) df.to_csv('results.csv', index=False)