def test_write_scen_file_world_only(tmpdir):
outfile = tmpdir.join("SCENARIO.SCEN")
write_scen_file(world_only, outfile)
output = read_scen_file(os.path.join(outfile.dirname, outfile.basename))
assert len(world_only) == len(output)
assert len(world_only.index) == len(output.index)
assert len(world_only.columns) == len(output.columns)
assert world_only.equals(output)
def test_write_scen_file(tmpdir):
outfile = tmpdir.join("SCENARIO.SCEN")
write_scen_file(rcp3pd, outfile)
output = read_scen_file(os.path.join(outfile.dirname, outfile.basename))
assert len(rcp3pd) == len(output)
assert len(rcp3pd["WORLD"].index) == len(output["WORLD"].index)
assert len(rcp3pd["WORLD"].columns) == len(output["WORLD"].columns)
assert rcp3pd["WORLD"].equals(output["WORLD"])
def test_read_world_only_scenario():
world_only = read_scen_file(
os.path.join(os.path.dirname(__file__), "./test_data/WORLD_ONLY.SCEN"))
assert isinstance(world_only, pd.DataFrame)
assert len(world_only) == 5
import os
import pandas as pd
import pytest
from pymagicc import (_magiccpath, _get_magicc_paths,
_get_number_of_datapoints, _get_region_code,
read_scen_file, write_scen_file, rcp26, rcp45, rcp60,
rcp85, run)
rcp26_scen_file = os.path.join(_magiccpath, "RCP26.SCEN")
rcp85_scen_file = os.path.join(_magiccpath, "RCP85.SCEN")
world_only = read_scen_file(
os.path.join(os.path.dirname(__file__), "./test_data/WORLD_ONLY.SCEN"))
def test_count():
assert _get_number_of_datapoints(rcp26_scen_file) == 20
def test_region_code():
assert _get_region_code(rcp26_scen_file) == 41
def test_read_scen_file():
assert len(rcp26) == 7
assert len(rcp26["WORLD"].index) == 20
assert len(rcp26["WORLD"].columns) == 23
def test_read_world_only_scenario():
world_only = read_scen_file(
def test_read_scen_file(mock_magicc_data):
result = read_scen_file(RCP26_SCEN_FILE)
mock_magicc_data.assert_called()
