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 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 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 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 __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 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 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()
