Пример #1
0
def test_db(request):
    tdf = IamDataFrame(request.cls.tdb)
    if request.param == "test_years":
        return tdf

    tdf = tdf.timeseries()
    tdf.columns = tdf.columns.map(lambda x: datetime(x, 6, 15))

    return IamDataFrame(tdf)
Пример #2
0
    def test_relationship_usage_wrong_time(self):
        tdb = IamDataFrame(self.tdb)
        tcruncher = self.tclass(tdb)
        res = tcruncher.derive_relationship("Emissions|CO2", ["Emissions|CO2"])

        test_downscale_df = IamDataFrame(self.tdb).timeseries()
        test_downscale_df.columns = test_downscale_df.columns.map(
            lambda x: dt.datetime(x, 1, 1))
        test_downscale_df = IamDataFrame(test_downscale_df)

        error_msg = re.escape(
            "`in_iamdf` time column must be the same as the time column used "
            "to generate this filler function (`year`)")
        with pytest.raises(ValueError, match=error_msg):
            res(test_downscale_df)