RustPython · youknowone · Sep 12, 2022 · Sep 9, 2022 · Sep 10, 2022 · Sep 10, 2022
diff --git a/Lib/operator.py b/Lib/operator.py
@@ -10,7 +10,7 @@
 This is the pure Python implementation of the module.
 """

-__all__ = ['abs', 'add', 'and_', 'attrgetter', 'concat', 'contains', 'countOf',
+__all__ = ['abs', 'add', 'and_', 'attrgetter', 'call', 'concat', 'contains', 'countOf',
           'delitem', 'eq', 'floordiv', 'ge', 'getitem', 'gt', 'iadd', 'iand',
           'iconcat', 'ifloordiv', 'ilshift', 'imatmul', 'imod', 'imul',
           'index', 'indexOf', 'inv', 'invert', 'ior', 'ipow', 'irshift',
@@ -221,6 +221,12 @@ def length_hint(obj, default=0):
        raise ValueError(msg)
    return val

+# Other Operations ************************************************************#
+
+def call(obj, /, *args, **kwargs):
+    """Same as obj(*args, **kwargs)."""
+    return obj(*args, **kwargs)
+
 # Generalized Lookup Objects **************************************************#

 class attrgetter:
@@ -423,6 +429,7 @@ def ixor(a, b):
 __abs__ = abs
 __add__ = add
 __and__ = and_
+__call__ = call
 __floordiv__ = floordiv
 __index__ = index
 __inv__ = inv

diff --git a/Lib/test/test_operator.py b/Lib/test/test_operator.py
@@ -45,6 +45,18 @@ def __iter__(self):


 class OperatorTestCase:
+    def test___all__(self):
+        operator = self.module
+        actual_all = set(operator.__all__)
+        computed_all = set()
+        for name in vars(operator):
+            if name.startswith('__'):
+                continue
+            value = getattr(operator, name)
+            if value.__module__ in ('operator', '_operator'):
+                computed_all.add(name)
+        self.assertSetEqual(computed_all, actual_all)
+
    def test_lt(self):
        operator = self.module
        self.assertRaises(TypeError, operator.lt)
@@ -518,6 +530,18 @@ def __length_hint__(self):
        with self.assertRaises(LookupError):
            operator.length_hint(X(LookupError))

+    def test_call(self):
+        operator = self.module
+
+        def func(*args, **kwargs): return args, kwargs
+
+        self.assertEqual(operator.call(func), ((), {}))
+        self.assertEqual(operator.call(func, 0, 1), ((0, 1), {}))
+        self.assertEqual(operator.call(func, a=2, obj=3),
+                         ((), {"a": 2, "obj": 3}))
+        self.assertEqual(operator.call(func, 0, 1, a=2, obj=3),
+                         ((0, 1), {"a": 2, "obj": 3}))
+
    def test_dunder_is_original(self):
        operator = self.module


diff --git a/vm/src/stdlib/operator.rs b/vm/src/stdlib/operator.rs
@@ -139,7 +139,7 @@ mod _operator {

    /// Return a @ b
    #[pyfunction]
-    fn mat_mul(a: PyObjectRef, b: PyObjectRef, vm: &VirtualMachine) -> PyResult {
+    fn matmul(a: PyObjectRef, b: PyObjectRef, vm: &VirtualMachine) -> PyResult {
        vm._matmul(&a, &b)
    }