def test_add(self):
buffer = tf.EventVectorBuffer()
self.assertEqual(len(buffer), 0)
buffer.append(
tf.AccessEvent(1, 1, 1, tf.AccessType.LOAD,
tf.MemoryLevel.MEM_LVL_HIT))
self.assertEqual(len(buffer), 1)
def test_write_read(self):
path = "./foo.txt"
a1 = tf.AccessEvent(1, 1, 42, tf.AccessType.LOAD,
tf.MemoryLevel.MEM_LVL_L1)
a2 = tf.AccessEvent(2, 2, 44, tf.AccessType.STORE,
tf.MemoryLevel.MEM_LVL_L2)
write_buffer = tf.EventVectorBuffer()
write_buffer.append(a1)
write_buffer.append(a2)
md = tf.TraceMetaData(write_buffer, 100)
with tf.TraceFile(path, tf.TraceFileMode.WRITE) as file:
self.assertEqual(path, file.path())
file.write(write_buffer, md)
self.assertTrue(os.path.isfile(path))
with tf.TraceFile(path, tf.TraceFileMode.READ) as file:
self.assertEqual(path, file.path())
read_buffer, read_md = file.read()
self.assertEqual(md.size(), read_md.size())
self.assertEqual(md.thread_id(), read_md.thread_id())
for expect, current in zip(write_buffer, read_buffer):
self.assertEqual(expect.timestamp, current.timestamp)
self.assertEqual(expect.address, current.address)
self.assertEqual(expect.ip, current.ip)
self.assertEqual(expect.type, current.type)
self.assertEqual(expect.level, current.level)
def test_set_item(self):
a1 = tf.AccessEvent(1, 1, 42, tf.AccessType.LOAD,
tf.MemoryLevel.MEM_LVL_L1)
a2 = tf.AccessEvent(2, 2, 44, tf.AccessType.STORE,
tf.MemoryLevel.MEM_LVL_L2)
buffer = tf.EventVectorBuffer()
buffer.append(a1)
buffer[0] = a2
self.assertEqual(a2.timestamp, buffer[0].timestamp)
self.assertEqual(a2.address, buffer[0].address)
self.assertEqual(a2.ip, buffer[0].ip)
self.assertEqual(a2.type, buffer[0].type)
self.assertEqual(a2.level, buffer[0].level)
def test_range(self):
buffer = tf.EventVectorBuffer()
data = [
tf.AccessEvent(1, 1, 1, tf.AccessType.LOAD,
tf.MemoryLevel.MEM_LVL_HIT),
tf.AccessEvent(2, 2, 2, tf.AccessType.STORE,
tf.MemoryLevel.MEM_LVL_MISS),
tf.AccessEvent(3, 3, 3, tf.AccessType.EXEC,
tf.MemoryLevel.MEM_LVL_L1),
]
for a in data:
buffer.append(a)
self.assertEqual(len(buffer), 3)
for a, e in zip(buffer, data):
self.assertEqual(a.timestamp, e.timestamp)
self.assertEqual(a.address, e.address)
self.assertEqual(a.ip, e.ip)
self.assertEqual(a.type, e.type)
self.assertEqual(a.level, e.level)
def test_creation(self):
buffer = tf.EventVectorBuffer()
md = tf.TraceMetaData(buffer, 1337)
self.assertEqual(md.size(), 0)
self.assertEqual(md.thread_id(), 1337)