stats: fix compilation of unit test.

This commit is contained in:
Ali Saidi 2012-05-10 18:04:28 -05:00
parent ec50c78f83
commit 331696582f
2 changed files with 5 additions and 2 deletions

View file

@ -237,7 +237,7 @@ class UnitTest(object):
'''Create a UnitTest'''
all = []
def __init__(self, target, *sources):
def __init__(self, target, *sources, **kwargs):
'''Specify the target name and any sources. Sources that are
not SourceFiles are evalued with Source(). All files are
guarded with a guard of the same name as the UnitTest
@ -252,6 +252,7 @@ class UnitTest(object):
self.sources = srcs
self.target = target
self.main = kwargs.get('main', False)
UnitTest.all.append(self)
# Children should have access
@ -906,6 +907,8 @@ def makeEnv(label, objsfx, strip = False, **kwargs):
flags = { test.target : True }
test_sources = Source.get(**flags)
test_objs = [ make_obj(s, static=True) for s in test_sources ]
if test.main:
test_objs += main_objs
testname = "unittest/%s.%s" % (test.target, label)
new_env.Program(testname, test_objs + static_objs)

View file

@ -52,7 +52,7 @@ UnitTest('trietest', 'trietest.cc')
stattest_py = PySource('m5', 'stattestmain.py', skip_lib=True)
stattest_swig = SwigSource('m5.internal', 'stattest.i', skip_lib=True)
UnitTest('stattest', 'stattest.cc', stattest_py, stattest_swig)
UnitTest('stattest', 'stattest.cc', stattest_py, stattest_swig, main=True)
UnitTest('symtest', 'symtest.cc')
UnitTest('tokentest', 'tokentest.cc')