def test_has_fields():
gdata = ge.GraphData(DF1)
assert gdata.has_field(["D"])
assert not gdata.has_field(["bla"])
gdata = ge.GraphData(DF2bis)
assert gdata.has_field(["NG"])
assert gdata.has_field(["NG"], region="A")
assert not gdata.has_field(["NG"], region="B")
def test_get_cols():
gdata = ge.GraphData(df=DF1)
assert gdata.get_cols() == list(gdata.df.columns)
assert gdata.get_cols(region="A") == ["CO2_A_D"]
assert gdata.get_cols(field="D") == ["CO2_A_D", "CO2_B_D"]
assert gdata.get_cols(region=["A"], field=["D"]) == ["CO2_A_D"]
gdata = ge.GraphData(df=DF2)
assert gdata.get_cols(region="A") == ["CO2_A_D", "CO2_A-B_ID"]
assert gdata.get_cols(field="ID") == ["CO2_A-B_ID"]
assert gdata.get_cols(region="B", field="ID") == []
gdata = ge.GraphData(df=DF3)
assert gdata.get_cols(region="A") == ["CO2_A_D", "CO2_A-B_ID"]
assert gdata.get_cols(field="ID") == ["CO2_A-B_ID", "CO2_B-A_ID"]
assert gdata.get_cols(region="B", field="ID") == ["CO2_B-A_ID"]
def test_check_positive(caplog):
gdata = ge.GraphData(df=DF4)
gdata.check_positive("A")
gdata.check_positive("C")
assert "A: 1 <0 for D" not in caplog.text
assert "C: 1 <0 for NG" in caplog.text
assert "C: 1 <0 for SRC_COL" in caplog.text
def test_parse_info2(caplog):
gdata = ge.GraphData(df=DF3)
assert gdata.regions == ["A", "B"]
assert gdata.region_fields == ["D"]
assert gdata.variable == "CO2"
assert gdata.link_fields == ["ID"]
assert gdata.fields == ["D", "ID"]
assert gdata.partners == {"A": ["B"], "B": ["A"]}
assert ("Inconsistencies in trade reporting - DEBUG has missing links"
not in caplog.text)
def test_parse_info3(caplog):
ge.configure_logging("DEBUG")
gdata = ge.GraphData(DF4)
assert gdata.regions == ["A", "B", "C", "D"]
assert gdata.region_fields == ["D", "NG", "TI", "SRC_BIO", "SRC_COL"]
assert gdata.variable == "E"
assert gdata.link_fields == ["ID"]
assert gdata.fields == ["D", "NG", "TI", "SRC_BIO", "SRC_COL", "ID"]
assert gdata.partners == {"A": ["B"], "B": ["A"], "C": ["D"], "D": ["C"]}
assert ("Inconsistencies in trade reporting - DEBUG has missing links"
not in caplog.text)
def test_parse_info(caplog):
gdata = ge.GraphData(df=DF1)
assert gdata.variable == "CO2"
assert gdata.regions == ["A", "B"]
assert gdata.region_fields == ["D"]
assert gdata.link_fields == []
assert gdata.fields == ["D"]
assert gdata.partners == {"A": [], "B": []}
assert ("Inconsistencies in trade reporting - DEBUG has missing links"
not in caplog.text)
ge.configure_logging("DEBUG")
gdata = ge.GraphData(df=DF2)
assert gdata.regions == ["A", "B"]
assert gdata.region_fields == ["D"]
assert gdata.variable == "CO2"
assert gdata.link_fields == ["ID"]
assert gdata.fields == ["D", "ID"]
assert gdata.partners == {"A": ["B"], "B": []}
assert "Inconsistencies in trade reporting - DEBUG has missing links" in caplog.text
assert "B-A" in caplog.text
def test_get_data():
gdata = ge.GraphData(df=DF1)
# check that get_data squeezes output correctly
print(DF1.loc[:, "CO2_A_D"])
print(gdata.get_data(region="A", field="D"))
tm.assert_series_equal(DF1.loc[:, "CO2_A_D"],
gdata.get_data(region="A", field="D"))
# No arguments should return the entire dataset
tm.assert_frame_equal(DF1, gdata.get_data())
# There is only one field in DF1, so this should also return the entire dataset
tm.assert_frame_equal(DF1, gdata.get_data(field="D"))
def test_parse_info5():
with pytest.raises(AssertionError):
gdata = ge.GraphData(DF2ter)
def test_parse_info4(caplog):
gdata = ge.GraphData(DF2bis)
assert "Regions for NG do not match overall regions!" in caplog.text
assert "Regions for ID do not match overall regions!" in caplog.text
def test_empty():
with pytest.raises(AttributeError):
ge.GraphData(pd.DataFrame())
def test_check_generation_by_source(caplog):
gdata = ge.GraphData(df=DF4)
gdata.check_generation_by_source("A")
gdata.check_generation_by_source("C")
assert "A: 1 NG != sum(NG_SRC)" in caplog.text
assert "C: 1 NG != sum(NG_SRC)" not in caplog.text
def test_check_antisymmetric(caplog):
gdata = ge.GraphData(df=DF4)
gdata.check_antisymmetric("A")
gdata.check_antisymmetric("C")
assert "A-B: 1 ID[i,j] != -ID[j,i]" not in caplog.text
assert "C-D: 1 ID[i,j] != -ID[j,i]" in caplog.text
def test_check_interchange(caplog):
gdata = ge.GraphData(df=DF4)
gdata.check_interchange("A")
gdata.check_interchange("C")
assert "A: 1 TI != sum(ID)" in caplog.text
assert "C: 1 TI != sum(ID)" not in caplog.text
def test_check_balance(caplog):
gdata = ge.GraphData(DF4)
gdata.check_balance("A")
gdata.check_balance("B")
assert "A: 1 TI+D != NG" in caplog.text
assert "B: 1 TI+D != NG" not in caplog.text
def test_check_nans1(caplog):
gdata = ge.GraphData(DF1)
gdata.check_nans("A")
gdata.check_nans("B")
assert "A: 1 NaNs for D" in caplog.text
assert "B: 1 NaNs for D" not in caplog.text
def test_malformed_columns(caplog):
gdata = ge.GraphData(df=DF2)
assert "Regions for ID do not match overall regions!" in caplog.text
