def test_exec():
rc = {
"exec": [
{
"file": os.path.join(TEST_DATA_DIR, "exec.py"),
"functions": ["do_exec"]
},
]
}
run_control().update(rc)
df = IamDataFrame(TEST_DF)
exp = ["bar"] * len(TEST_DF["scenario"].unique())
obs = df["foo"].values
assert (exp == obs).all()
def test_exec():
rc = {
'exec': [
{
'file': os.path.join(TEST_DATA_DIR, 'exec.py'),
'functions': ['do_exec']
},
]
}
run_control().update(rc)
df = IamDataFrame(TEST_DF)
exp = ['bar'] * len(TEST_DF['scenario'].unique())
obs = df['foo'].values
assert ((exp == obs).all())
def update_run_control(update):
run_control().update(update)
yield
reset_rc_defaults()
#%%
pyam.categorize(sr1p5,
exclude=False,
subcategory='uncategorized',
value='Above 2C',
name='subcategory',
criteria={warming_exccedance_prob(2.0): {
'up': 1.0
}},
color='darkgrey')
#%% [markdown]
# ### Aggregation of subcategories to categories
#%%
rc = pyam.run_control()
def assign_rc_color_from_sub(cat, sub):
rc.update({'color': {'category': {cat: rc['color']['subcategory'][sub]}}})
#%%
sr1p5.set_meta(
meta='Below 1.5C',
name='category',
index=sr1p5.filter(
subcategory=['Below 1.5C (I)', 'Below 1.5C (II)']).meta.index)
assign_rc_color_from_sub('Below 1.5C', 'Below 1.5C (II)')
#%%
def test_no_file():
rc = run_control()
pytest.raises(IOError, rc.update, "no_such_file.yaml")
import pandas as pd
import pyam
import matplotlib.pyplot as plt
import numpy as np
import copy
from matplotlib.patches import Patch
from utils import *
PLOT_KWARGS = {'lw': 3}
pyam.run_control().update('./plotting.yaml')
legend = dict(loc='center left', bbox_to_anchor=(1.0, 0.5), prop={'size': 16})
col = 'current-region-defs'
col = 'NEW-AR6-Ch6-Fig6.4'
regions = sorted(
pd.read_csv('country_mapping_ISO-5Regions.csv')
[col]
.unique()
)
# gas order
gases = ['CH4', 'BC', 'OC', 'Sulfur', 'NOx', 'CO', 'VOC', 'NH3']
units = [
r'Tg (CH$_4$) yr$^{-1}$',
r'Tg (C) yr$^{-1}$',
r'Tg (C) yr$^{-1}$',
r'Tg (SO$_2$) yr$^{-1}$',
r'Tg (NO$_2$) yr$^{-1}$',
