def test_big_nid_map_scalable(self): """check big nid mapping is scalable.""" model = Model() model.parse("nid_map: nodes=foo[1-9999] nids=bar[1-9999]@tcp") before = time.time() NidMap.fromlist(model.get("nid_map")) self.assertTrue(time.time() - before < 0.5)
def testMultipleNid(self): """add multiple nid per node""" nm = NidMap() nm.add("foo[4-5]", "foo[4-5]-ib0@o2ib1") nm.add("foo[4-5]", "foo[4-5]-ib1@o2ib2") self.assertEqual(nm["foo4"], ["foo4-ib0@o2ib1", "foo4-ib1@o2ib2"]) self.assertEqual(nm["foo5"], ["foo5-ib0@o2ib1", "foo5-ib1@o2ib2"])
def test_big_nid_map_scalable(self): """check big nid mapping is scalable.""" model = Model() model.parse("nid_map: nodes=foo[1-9999] nids=bar[1-9999]@tcp") before = time.time() NidMap.fromlist(model.get('nid_map')) self.assertTrue(time.time() - before < .5)
def testFromList(self): """construct from a list of modelnidmap""" line1 = ModelNidMap() line1.parse("nodes=foo[1-2] nids=foo[1-2]-ib0@o2ib4") line2 = ModelNidMap() line2.parse("nodes=bar[1-2] nids=bar[1-2]-ib0@o2ib4") nm = NidMap.fromlist([line1, line2]) self.assertEqual(nm["foo1"], ["foo1-ib0@o2ib4"]) self.assertEqual(nm["foo2"], ["foo2-ib0@o2ib4"]) self.assertEqual(nm["bar1"], ["bar1-ib0@o2ib4"]) self.assertEqual(nm["bar2"], ["bar2-ib0@o2ib4"])
def __init__(self, filename): self.backend = None self.xmf_path = None self.model = Model() try: self.model.load(filename) except IOError: raise ModelFileIOError("Could not read %s" % filename) # Set nodes to nids mapping using the NidMap helper class self.nid_map = NidMap.fromlist(self.get('nid_map')) # Initialize the tuning model to None if no special tuning configuration # is provided self.tuning_model = TuningModel()
def testAddMapping(self): """add() method""" nm = NidMap() nm.add("bar[1-3]", "bar[1-3]-ib0@o2ib3") self.assertEqual(nm["bar1"], ["bar1-ib0@o2ib3"]) self.assertEqual(nm["bar2"], ["bar2-ib0@o2ib3"])
def testString(self): """cast to string is ok""" nm = NidMap("foo[1-2]", "foo[1-2]@o2ib2") str(nm)
def testSimple(self): """simple mapping""" nm = NidMap("foo[1-2]", "foo[1-2]@o2ib2") self.assertEqual(nm["foo1"], ["foo1@o2ib2"]) self.assertEqual(nm["foo2"], ["foo2@o2ib2"])
def testEmpty(self): """empty nidmap""" nm = NidMap()