Initialize repository

Author: atpalmer

.gitignore

@@ -0,0 +1,5 @@
+.cache/
+.venv/
+
+__pycache__/
+*.pyc

setup.py

@@ -0,0 +1,11 @@
+from setuptools import setup, Extension
+
+
+setup(
+    name='monkeypatch',
+    ext_modules=[
+        Extension('monkeypatch', sources=[
+            'src/monkeypatch.c',
+        ]),
+    ],
+)

src/monkeypatch.c

@@ -0,0 +1,31 @@
+#include "Python.h"
+
+static PyObject *monkeypatch(PyObject *self, PyObject *args) {
+    PyObject *target;
+    PyObject *attr_name;
+    PyObject *attr_val;
+
+    if(!PyArg_ParseTuple(args, "OOO", &target, &attr_name, &attr_val))
+        return NULL;
+
+    _PyObject_GenericSetAttrWithDict(target, attr_name, attr_val, NULL);
+
+    Py_RETURN_NONE;
+}
+
+struct PyMethodDef methods[] = {
+    { "monkeypatch", monkeypatch, METH_VARARGS, "" },
+    { 0 },
+};
+
+struct PyModuleDef module = {
+    .m_base = PyModuleDef_HEAD_INIT,
+    .m_name = "monkeypatch",
+    .m_doc = "",
+    .m_size = 0,
+    .m_methods = methods,
+};
+
+PyMODINIT_FUNC PyInit_monkeypatch(void) {
+    PyModule_Create(&module);
+}

test/test_monkeypatch.py

@@ -0,0 +1,30 @@
+import pytest
+from monkeypatch import monkeypatch
+
+
+def test_monkeypatch_hello():
+    with pytest.raises(TypeError):
+        str.hello = "world"
+    monkeypatch(str, "hello", "world")
+    assert str.hello == "world"
+
+
+def test_monkeypatch_str():
+    def rev_up(self):
+        return self[::-1].upper()
+    monkeypatch(str, "rev_up", rev_up)
+    assert "abc".rev_up() == "CBA"
+
+
+def test_monkeypatch_int():
+    def upto(self, value):
+        return tuple(range(self, value + 1))
+    monkeypatch(int, "upto", upto)
+    assert (1).upto(10) == (1,2,3,4,5,6,7,8,9,10)
+
+
+def test_monkeypatch_tuple():
+    def mult_by(self, value):
+        return tuple(x * value for x in self)
+    monkeypatch(tuple, "mult_by", mult_by)
+    assert (1,2,3).mult_by(10) == (10,20,30)