Source code for dataprocessor.tests.test_configure

# coding: utf-8

import os.path as op
import unittest
from ..utility import check_directory
from ..pipes.configure import load


ROOT = op.join(__file__, "../../../../sample/datadir")


[docs]class TestConfigure_INI(unittest.TestCase):
[docs] def setUp(self): self.node_list = [{ "path": check_directory(op.join(ROOT, "project2/run01")), "type": "run", }]
[docs] def test_load(self): nl = load(self.node_list, "parameters.ini") self.assertEqual(nl[0]["configure"], {"ny": "23"}) nl = load(nl, "parameters.ini") # works if there already exist "configure" self.assertEqual(nl[0]["configure"], {"ny": "23"})
[docs] def test_load_filetype(self): nl = load(self.node_list, "parameters.ini", filetype="yaml") self.assertNotIn("configure", nl[0]) # fail to parse YAML nl = load(self.node_list, "parameters.ini", filetype="INI") self.assertEqual(nl[0]["configure"], {"ny": "23"})
[docs] def test_load_missing_name(self): load(self.node_list, "parame.yamlu") # not raises
[docs]class TestConfigure_YAML(unittest.TestCase):
[docs] def setUp(self): self.node_list = [{ "path": check_directory(op.join(ROOT, "project3/run01")), "type": "run", }]
[docs] def test_load(self): nl = load(self.node_list, "parameters.yml") self.assertEqual(nl[0]["configure"], {"ny": 23}) nl = load(nl, "parameters.yml") self.assertEqual(nl[0]["configure"], {"ny": 23})
[docs] def test_load_filetype_yaml(self): nl = load(self.node_list, "parameters.yml", filetype="INI") self.assertNotIn("configure", nl[0]) # fail to parse YAML nl = load(self.node_list, "parameters.yml", filetype="YAML") self.assertEqual(nl[0]["configure"], {"ny": 23})
[docs]class TestConfigure_CONF(unittest.TestCase):
[docs] def setUp(self): self.node_list = [{ "path": check_directory(op.join(ROOT, "project1/run01")), "type": "run", }]
[docs] def test_load_conf(self): nl = load(self.node_list, "parameters.conf") self.assertEqual(nl[0]["configure"]["nx"], "12") self.assertEqual(nl[0]["configure"]["ny"], "32")