StabilityMatrix/StabilityMatrix.Tests/Python/PyRunnerTests.cs

using Python.Runtime;
using StabilityMatrix.Core.Python;

namespace StabilityMatrix.Tests.Python;

[TestClass]
public class PyRunnerTests
{
    private static readonly PyRunner PyRunner = new();
    
    [ClassInitialize]
    public static async Task TestInitialize(TestContext testContext)
    {
        var appData = Environment.GetFolderPath(Environment.SpecialFolder.ApplicationData);
        PyRunner.HomeDir = Path.Combine(appData, "StabilityMatrix");
        await PyRunner.Initialize();
    }
    
    [TestMethod]
    public void PythonEngine_ShouldBeInitialized()
    {
        Assert.IsTrue(PythonEngine.IsInitialized);
    }
    
    [TestMethod]
    public async Task RunEval_ShouldReturnOutput()
    {
        // Arrange
        const string script = "print('Hello World')";
        
        // Act
        var result = await PyRunner.Eval(script);
        var stdout = PyRunner.StdOutStream!.GetBuffer();
        
        // Assert
        Assert.AreEqual("Hello World\n", stdout);
        Assert.AreEqual("None", result);
    }
}