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_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_global_config_parsing(self):
ConfigParser.parse(join(_basedir, "test_data/test.yaml"))
self.assertEqual(_global_data["NoOfPrimaries"], 10.)
def test_falseParse(self):
d = ConfigParser.parse(join(_basedir, "test_data/test.yaml"))
self.assertNotEqual(d, self.data2)
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 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)
def test_parseOrder(self):
od = ConfigParser.parse(join(_basedir, "test_data/test2.yaml"))
res = [1, 3, 2]
self.assertEqual(od['plugins'].keys(), res)