def setUp(self):
self.process = MagicMock()
self.child = MagicMock()
def getitem(name):
return name
self.child.__getitem__.side_effect = getitem
self.params = MagicMock()
self.params.mri = "BLOCK-STAT"
self.process.get_block.return_value = self.child
self.o = StatsPluginPart(self.process, self.params)
class TestStatsPluginPart(unittest.TestCase):
def setUp(self):
self.process = MagicMock()
self.child = MagicMock()
def getitem(name):
return name
self.child.__getitem__.side_effect = getitem
self.params = MagicMock()
self.params.mri = "BLOCK-STAT"
self.process.get_block.return_value = self.child
self.o = StatsPluginPart(self.process, self.params)
def test_report_info(self):
infos = self.o.report_info(ANY)
self.assertEqual(len(infos), 1)
self.assertEqual(infos[0].name, "sum")
self.assertEqual(infos[0].attr, "StatsTotal")
def test_configure(self):
task = MagicMock()
completed_steps = ANY
steps_to_do = ANY
part_info = ANY
params = MagicMock()
params.filePath = "/tmp/anything.h5"
infos = self.o.configure(task, completed_steps, steps_to_do, part_info,
params)
self.assertIsNone(infos)
task.put_many_async.assert_called_once_with(
self.child, dict(enableCallbacks=True, computeStatistics=True))
expected_filename = "/tmp/BLOCK-STAT-attributes.xml"
task.put_async.assert_called_once_with(self.child["attributesFile"],
expected_filename)
expected_xml = """<?xml version="1.0" ?>
<Attributes>
<Attribute addr="0" datatype="DOUBLE" description="Sum of the array" name="StatsTotal" source="TOTAL" type="PARAM" />
</Attributes>"""
actual_xml = open(expected_filename).read().replace(">", ">\n")
self.assertEqual(actual_xml.splitlines(), expected_xml.splitlines())
def setUp(self):
self.process = MagicMock()
self.child = MagicMock()
def getitem(name):
return name
self.child.__getitem__.side_effect = getitem
self.params = MagicMock()
self.process.get_block.return_value = self.child
self.o = StatsPluginPart(self.process, self.params)
class TestStatsPluginPart(unittest.TestCase):
def setUp(self):
self.process = MagicMock()
self.child = MagicMock()
def getitem(name):
return name
self.child.__getitem__.side_effect = getitem
self.params = MagicMock()
self.process.get_block.return_value = self.child
self.o = StatsPluginPart(self.process, self.params)
def test_configure(self):
task = MagicMock()
completed_steps = ANY
steps_to_do = ANY
part_info = ANY
self.o.configure(task, completed_steps, steps_to_do, part_info)
task.put_many.assert_called_once_with(self.child, dict(
enableCallbacks=True,
computeStatistics=True))
