1 files changed, 46 insertions, 8 deletions
diff --git a/configs/test/test.py b/configs/test/test.py
index 75e832f5e..8c5b06e6a 100644
--- a/configs/test/test.py
+++ b/configs/test/test.py
@@ -1,12 +1,50 @@
-from m5 import *
+# Simple test script
+#
+# Alpha: "m5 test.py"
+# MIPS: "m5 test.py -a Mips -c hello_mips"
 
-class HelloWorld(AlphaLiveProcess):
-    executable = '../configs/test/hello'
-    cmd = 'hello'
+import os, optparse, sys
+import m5
+from m5.objects import *
+
+# parse command-line arguments
+parser = optparse.OptionParser(option_list=m5.standardOptions)
+
+parser.add_option("-c", "--cmd", default="hello")
+parser.add_option("-t", "--timing", action="store_true")
+
+(options, args) = parser.parse_args()
+
+if args:
+    print "Error: script doesn't take any positional arguments"
+    sys.exit(1)
+
+# build configuration
+this_dir = os.path.dirname(__file__)
+
+process = LiveProcess()
+process.executable = os.path.join(this_dir, options.cmd)
+process.cmd = options.cmd
 
 magicbus = Bus()
 mem = PhysicalMemory()
-cpu = AtomicSimpleCPU(workload=HelloWorld(), mem=magicbus)
-system = System(physmem=mem, cpu=cpu)
-system.c1 =  Connector(side_a=mem, side_b=magicbus)
-root = Root(system=system)
+
+if options.timing:
+    cpu = TimingSimpleCPU()
+else:
+    cpu = AtomicSimpleCPU()
+cpu.workload = process
+cpu.mem = magicbus
+
+system = System(physmem = mem, cpu = cpu)
+system.c1 =  Connector(side_a = mem, side_b = magicbus)
+root = Root(system = system)
+
+# instantiate configuration
+m5.instantiate(root)
+
+# simulate until program terminates
+exit_event = m5.simulate()
+
+print 'Exiting @', m5.curTick(), 'because', exit_event.getCause()
+