Minimal definition of dis.dis.

cthulahoops · cthulahoops · commit 2c0016a17171 · 2019-02-26T09:41:39.000Z
diff --git a/tests/snippets/dismod.py b/tests/snippets/dismod.py
@@ -1,10 +1,10 @@
 import dis
 
-dis.disassemble(compile("5 + x + 5 or 2", "", "eval"))
+dis.dis(compile("5 + x + 5 or 2", "", "eval"))
 print("\n")
-dis.disassemble(compile("def f(x):\n   return 1", "", "exec"))
+dis.dis(compile("def f(x):\n   return 1", "", "exec"))
 print("\n")
-dis.disassemble(compile("if a:\n 1 or 2\nelif x == 'hello':\n 3\nelse:\n 4", "", "exec"))
+dis.dis(compile("if a:\n 1 or 2\nelif x == 'hello':\n 3\nelse:\n 4", "", "exec"))
 print("\n")
-dis.disassemble(compile("f(x=1, y=2)", "", "eval"))
+dis.dis(compile("f(x=1, y=2)", "", "eval"))
 print("\n")
diff --git a/vm/src/stdlib/dis.rs b/vm/src/stdlib/dis.rs
@@ -2,6 +2,10 @@ use crate::obj::objcode;
 use crate::pyobject::{PyContext, PyFuncArgs, PyObjectRef, PyResult, TypeProtocol};
 use crate::vm::VirtualMachine;
 
+fn dis_dis(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
+    dis_disassemble(vm, args)
+}
+
 fn dis_disassemble(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
     arg_check!(vm, args, required = [(co, Some(vm.ctx.code_type()))]);
 
@@ -12,6 +16,7 @@ fn dis_disassemble(vm: &mut VirtualMachine, args: PyFuncArgs) -> PyResult {
 
 pub fn mk_module(ctx: &PyContext) -> PyObjectRef {
     py_module!(ctx, "dis", {
+        "dis" => ctx.new_rustfunc(dis_dis),
         "disassemble" => ctx.new_rustfunc(dis_disassemble)
     })
 }
