runtests.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

#!/usr/bin/env python3

###############################################################################
# Description
###############################################################################

# Unit Tests


###############################################################################
# Imports
###############################################################################

import trace_reader
import naive_trace_reader
import os

###############################################################################
# Trace files
###############################################################################

files = [
'./sample_traces/st-g6-0.bin',
'./sample_traces/st-g6-1.bin',
'./sample_traces/st-g6-2.bin',
'./sample_traces/st-g6-3.bin',
]

###############################################################################
# Tests
###############################################################################

# Does our fancy trace reader get the same number of files as our naive one?
# (See naive_trace_reader.py for further explanation)
def test1():
    stream = trace_reader.trace_reader(files)
    num_records = len(list(stream))
    stream = naive_trace_reader.trace_reader(files)
    naive_num_records = len(list(stream))

    # We need a +1 here because the fancy reader produces a 'meta' record
    # indicating the number of CPUs
    if num_records != naive_num_records + 1:
        return "[FAIL]"
    return "[SUCCESS]"

print("Test 1: {}".format(test1()))