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}$',