Convert None dict keys to python.none in "table_from()" (GH-223)

Previously resulted in a crash. Closes #137
scoder · Apr 3, 2023 · 2b569fb · 2b569fb
1 parent a1383f3
commit 2b569fb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/lupa/_lupa.pyx b/lupa/_lupa.pyx
@@ -430,7 +430,7 @@ cdef class LuaRuntime:
             for obj in args:
                 if isinstance(obj, dict):
                     for key, value in obj.iteritems():
-                        py_to_lua(self, L, key)
+                        py_to_lua(self, L, key, wrap_none=True)
                         py_to_lua(self, L, value)
                         lua.lua_rawset(L, -3)
 
@@ -447,7 +447,7 @@ cdef class LuaRuntime:
                 elif isinstance(obj, Mapping):
                     for key in obj:
                         value = obj[key]
-                        py_to_lua(self, L, key)
+                        py_to_lua(self, L, key, wrap_none=True)
                         py_to_lua(self, L, value)
                         lua.lua_rawset(L, -3)
                 else:

diff --git a/lupa/tests/test.py b/lupa/tests/test.py
@@ -494,10 +494,11 @@ def test_create_table_args_kwargs(self):
         self.assertEqual(6, len(table))
 
     def test_table_from_dict(self):
-        table = self.lua.table_from({"foo": 1, "bar": 20, "baz": "spam"})
+        table = self.lua.table_from({"foo": 1, "bar": 20, "baz": "spam", None: "python.none"})
         self.assertEqual(     1, table['foo'])
         self.assertEqual(    20, table['bar'])
         self.assertEqual("spam", table['baz'])
+        self.assertEqual("python.none", table[None])
 
         self.assertEqual(0, len(table))