def test_filter():
ppl = Ppl(TEST_FLD + "FC1_rev01.ppl")
PTs = ppl.filter_data('PT')
assert 'PT' in PTs[4]
assert 'old_offshore' in PTs[4]
ppl.profiles
assert 'GG' in ppl.profiles[1]
def test_init():
ppl = Ppl(TEST_FLD + "FC1_rev01.ppl")
assert ppl.fname == "FC1_rev01.ppl"
assert ppl._attributes['branch_idx'][0] == 18
branch = 'tiein_spool'
assert int(ppl.geometries[branch][0][0]) == 0
assert int(ppl.geometries[branch][0][-1]) == 265
assert int(ppl.geometries[branch][1][0]) == -120
assert int(ppl.geometries[branch][1][11]) == -120
def test_to_excel():
ppl = Ppl("FC1_rev01.ppl")
ppl.to_excel()
assert "FC1_rev01_ppl.xlsx" in os.listdir()
xl = xlrd.open_workbook("FC1_rev01_ppl.xlsx")
sh = xl.sheet_by_index(14)
assert sh.cell_value(2, 2) == 1.654940e1
os.remove("FC1_rev01_ppl.xlsx")
ppl.to_excel("/tmp")
assert "FC1_rev01_ppl.xlsx" in os.listdir("/tmp")
os.remove("/tmp/FC1_rev01_ppl.xlsx")
def test_to_excel():
ppl = Ppl(TEST_FLD + "FC1_rev01.ppl")
ppl.to_excel()
assert "FC1_rev01_ppl.xlsx" in os.listdir()
xl = xlrd.open_workbook("FC1_rev01_ppl.xlsx")
sh = xl.sheet_by_index(14)
assert sh.cell_value(2, 2) == 1.654940e1
os.remove("FC1_rev01_ppl.xlsx")
temp_folder = tempfile.gettempdir()
ppl.to_excel(temp_folder)
assert "FC1_rev01_ppl.xlsx" in os.listdir(temp_folder)
os.remove(temp_folder + os.sep + "FC1_rev01_ppl.xlsx")
def test_not_a_ppl():
with pytest.raises(ValueError) as exeinfo:
ppl = Ppl(TEST_FLD + "/FC1_rev01.tpl")
assert exinfo.value.message == "not a ppl file"
def test_extraction():
ppl = Ppl(TEST_FLD + "FC1_rev01.ppl")
ppl.extract(4)
assert ppl.data[4][1][0][0] == 9.962770e6
assert ppl.data[4][1][-1][0] == 1.276020e7
def test_attributes():
ppl = Ppl(TEST_FLD + "FC1_rev01.ppl")
assert ppl._attributes['CATALOG'] == 331
assert ppl._attributes['data_idx'] == 381
assert 'GG' in ppl.profiles[1]
assert ppl._attributes['nvar'] == 48
def test_time_series():
ppl = Ppl(TEST_FLD + "FC1_rev01.ppl")
assert int(ppl.time[0]) == 0
assert int(ppl.time[-1]) == 1.8e5
