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()