def test_weight_parsing(self): config = {"global": {"weights": {"foo": 0.1, "bar": 0.8}}, "plugins": {}} f_config_name = join(_basedir, "test_data/config_weights.yaml") f_config = open(f_config_name, "w") dump(config, f_config) f_config.close() res = {"weights": {"foo": 0.1, "bar": 0.8}} parsed_config = ConfigParser.parse(f_config_name) print _global_data.__dict__ self.assertEqual(_global_data, res) os.remove(f_config_name)
def test_detector_parsing_types_global(self): config = {"global": {"NoOfPrimaries": "float:10"}, "plugins": {"a": [1, 2, 3]}} f_config_name = join(_basedir, "test_data/config_type_global.yaml") f_config = open(f_config_name, "w") dump(config, f_config) f_config.close() res = {"NoOfPrimaries": 10.} config = ConfigParser.parse(f_config_name) self.assertEqual(_global_data, res) os.remove(f_config_name)
def testAttrParsingTable(self): config = ConfigParser.parse(join(_basedir, "test_data/testconfig.yaml")) res = { "plugins": { "AoverLECalculator": None, "TableMaker": { "cols": ["Isotope", "Activity", "AoverLE"] } } } self.assertEqual(config, res)
def testAttrParsingPlot(self): config = ConfigParser.parse( join(_basedir, "test_data/testconfig_plot.yaml")) res = { "plugins": { "PlotMaker": { "plots": { "activity": { "type": "2D", "quantity": "Activity" } } } } } self.assertEqual(config, res)
def test_detector_parsing_types_global(self): config = { "global": { "NoOfPrimaries": "float:10" }, "plugins": { "a": [1, 2, 3] } } f_config_name = join(_basedir, "test_data/config_type_global.yaml") f_config = open(f_config_name, "w") dump(config, f_config) f_config.close() res = {"NoOfPrimaries": 10.} config = ConfigParser.parse(f_config_name) self.assertEqual(_global_data, res) os.remove(f_config_name)
def test_weight_parsing(self): config = { "global": { "weights": { "foo": 0.1, "bar": 0.8 } }, "plugins": {} } f_config_name = join(_basedir, "test_data/config_weights.yaml") f_config = open(f_config_name, "w") dump(config, f_config) f_config.close() res = {"weights": {"foo": 0.1, "bar": 0.8}} parsed_config = ConfigParser.parse(f_config_name) print _global_data.__dict__ self.assertEqual(_global_data, res) os.remove(f_config_name)
def testBuildGraphNodesFromYAML(self): od = CP.parse(join(_basedir, "test_data/testconfig.yaml")) graph = GB.build_graph(od['plugins']) graphRef = ["Input", "AoverLECalculator", "TableMaker", "Output"] self.assertItemsEqual(graph.nodes(), graphRef)
def test_parseConfig(self): d = ConfigParser.parse(join(_basedir, "test_data/test.yaml")) res = {"plugins": {"a": [1, 2, 3]}} mass_res = {"det1": {"mass": Mass(100, ureg.kg)}, "NoOfPrimaries": 10} self.assertEqual(d, res) self.assertEqual(_global_data, mass_res)
def test_parseOrder(self): od = ConfigParser.parse(join(_basedir, "test_data/test2.yaml")) res = [1, 3, 2] self.assertEqual(od['plugins'].keys(), res)
def test_falseParse(self): d = ConfigParser.parse(join(_basedir, "test_data/test.yaml")) self.assertNotEqual(d, self.data2)
def test_global_config_parsing(self): ConfigParser.parse(join(_basedir, "test_data/test.yaml")) self.assertEqual(_global_data["NoOfPrimaries"], 10.)
def test_detector_parsing(self): res = {"det1": {"mass": Mass(100., ureg.kg)}, "NoOfPrimaries": 10} config = ConfigParser.parse(join(_basedir, "test_data/test.yaml")) self.assertEqual(_global_data, res)
def testAttrParsingPlot(self): config = ConfigParser.parse(join(_basedir, "test_data/testconfig_plot.yaml")) res = {"plugins": {"PlotMaker": {"plots": {"activity": {"type": "2D", "quantity": "Activity"}}}}} self.assertEqual(config, res)
def testAttrParsingTable(self): config = ConfigParser.parse(join(_basedir, "test_data/testconfig.yaml")) res = {"plugins": {"AoverLECalculator": None, "TableMaker": {"cols": ["Isotope", "Activity", "AoverLE"]}}} self.assertEqual(config, res)