fix transforms unittest

sbourdeauducq · sbourdeauducq · commit c98e08fe36fb · 2015-04-19T11:40:49.000+08:00
diff --git a/artiq/coredevice/comm_dummy.py b/artiq/coredevice/comm_dummy.py
@@ -6,8 +6,7 @@
 
 
 class _RuntimeEnvironment(LinkInterface):
-    def __init__(self, ref_period):
-        self.internal_ref_period = ref_period
+    def __init__(self):
         self.warmup_time = 1*ms
 
     def emit_object(self):
@@ -16,7 +15,7 @@ def emit_object(self):
 
 class Comm(AutoDB):
     def get_runtime_env(self):
-        return _RuntimeEnvironment(1*ns)
+        return _RuntimeEnvironment()
 
     def switch_clock(self, external):
         pass
diff --git a/artiq/test/transforms.py b/artiq/test/transforms.py
@@ -1,6 +1,7 @@
 import unittest
 import ast
 
+from artiq import ns
 from artiq.coredevice import comm_dummy, core
 from artiq.transforms.unparse import unparse
 
@@ -39,7 +40,7 @@ def run():
 
 class OptimizeCase(unittest.TestCase):
     def test_optimize(self):
-        coredev = core.Core(comm=comm_dummy.Comm())
+        coredev = core.Core(comm=comm_dummy.Comm(), ref_period=1*ns)
         func_def = ast.parse(optimize_in).body[0]
         coredev.transform_stack(func_def, dict(), dict())
         self.assertEqual(unparse(func_def), optimize_out)
