Fix some python3 problems

hibtc · May 4, 2014 · f61a821 · f61a821
1 parent d8dc3c9
commit f61a821
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/madseq.py b/madseq.py
@@ -581,7 +581,7 @@ def __call__(self, node, defs):
         """
 
         if isinstance(node, Element):
-            defs[node.name] = node
+            defs[str(node.name)] = node
             node._base = defs.get(node.type)
         if not isinstance(node, Sequence):
             return node

diff --git a/test/test_element.py b/test/test_element.py
@@ -1,7 +1,7 @@
 # test utilities
 import unittest
 
-from pydicti import odicti
+from pydicti import odicti, dicti
 
 # tested module
 import madseq
@@ -38,8 +38,8 @@ def test_all_args(self):
         el1 = madseq.Element(None, None, odicti(a='a1', b='b1', d='d1'), el0)
         el2 = madseq.Element(None, None, odicti(a='a2'), el1)
         self.assertEqual(el2.all_args,
-                         odicti([('c', 'c0'), ('b', 'b1'),
-                                 ('d', 'd1'), ('a', 'a2')]))
+                         dicti([('c', 'c0'), ('b', 'b1'),
+                                ('d', 'd1'), ('a', 'a2')]))
 
     def test_copy(self):
         el = madseq.Element(None, None, odicti(a=1))