pythonnet/src/embed_tests/CallableObject.cs at net8.0 · pythonnet/pythonnet · GitHub

History

87 lines (76 loc) · 2.69 KB

Raw

using System;

using System.Collections.Generic;

using NUnit.Framework;

using Python.Runtime;

namespace Python.EmbeddingTest

{

public class CallableObject

{

[OneTimeSetUp]

public void SetUp()

{

PythonEngine.Initialize();

using var locals = new PyDict();

PythonEngine.Exec(CallViaInheritance.BaseClassSource, locals: locals);

CustomBaseTypeProvider.BaseClass = new PyType(locals[CallViaInheritance.BaseClassName]);

PythonEngine.InteropConfiguration.PythonBaseTypeProviders.Add(new CustomBaseTypeProvider());

}

[OneTimeTearDown]

public void Dispose()

{

PythonEngine.Shutdown();

}

[Test]

public void CallMethodMakesObjectCallable()

{

var doubler = new DerivedDoubler();

dynamic applyObjectTo21 = PythonEngine.Eval("lambda o: o(21)");

Assert.AreEqual(doubler.__call__(21), (int)applyObjectTo21(doubler.ToPython()));

}

[Test]

public void CallMethodCanBeInheritedFromPython()

{

var callViaInheritance = new CallViaInheritance();

dynamic applyObjectTo14 = PythonEngine.Eval("lambda o: o(14)");

Assert.AreEqual(callViaInheritance.Call(14), (int)applyObjectTo14(callViaInheritance.ToPython()));

}

[Test]

public void CanOverwriteCall()

{

var callViaInheritance = new CallViaInheritance();

using var scope = Py.CreateScope();

scope.Set("o", callViaInheritance);

scope.Exec("orig_call = o.Call");

scope.Exec("o.Call = lambda a: orig_call(a*7)");

int result = scope.Eval<int>("o.Call(5)");

Assert.AreEqual(105, result);

}

class Doubler

{

public int __call__(int arg) => 2 * arg;

}

class DerivedDoubler : Doubler { }

class CallViaInheritance

{

public const string BaseClassName = "Forwarder";

public static readonly string BaseClassSource = $@"

class MyCallableBase:

def __call__(self, val):

return self.Call(val)

class {BaseClassName}(MyCallableBase): pass

public int Call(int arg) => 3 * arg;

}

class CustomBaseTypeProvider : IPythonBaseTypeProvider

{

internal static PyType BaseClass;

public IEnumerable<PyType> GetBaseTypes(Type type, IList<PyType> existingBases)

{

Assert.Greater(BaseClass.Refcount, 0);

return type != typeof(CallViaInheritance)

? existingBases

: new[] { BaseClass };

}

pythonnet/src/embed_tests/CallableObject.cs at net8.0 · pythonnet/pythonnet · GitHub

Uh oh!

Expand file tree

Breadcrumbs

CallableObject.cs

Latest commit

History

Breadcrumbs

CallableObject.cs

File metadata and controls