Function TEST_P

test/test_python_plugin_manager.cpp:49–218

};

TEST_P(TestPythonPlugin, PythonPlugin) {
    using namespace lgraph;
    using namespace python_plugin;
    std::string plugin_dir = "./python_plugin";
    size_t n_workers = 2, n_jobs = 3;
    n_workers = GetParam().j;
    n_jobs = GetParam().n;
    int argc = _ut_argc;
    char** argv = _ut_argv;
    Configuration config;
    config.Add(plugin_dir, "d,dir", true).Comment("Directory where plugin files are stored.");
    config.Add(n_workers, "j,workers", true).Comment("Number of Python worker threads to use.");
    config.Add(n_jobs, "n,jobs", true).Comment("Number of Python jobs.");
    config.ParseAndFinalize(argc, argv);
    const char* sleep_plugin = "./sleep.py";
    const char* read_plugin = "./scan_graph.py";
    const char* write_plugin = "./write_vertex.py";
    std::string code_sleep;
    std::string code_read;
    std::string code_write;
    {
        UT_LOG() << "Create test files";
        code_sleep = R"(
import time
def Process(db, input):
    t = 1
    try:
        t = int(input)
    except:
        return (False, 'input must be a time duration')
    print('sleeping for {} seconds'.format(t))
    time.sleep(t)
    return (True, '')
)";
        code_read = R"(
def Process(db, input):
    n = 0
    try:
        n = int(input)
    except:
        pass
    if n == 0:
        n = 1000000
    txn = db.CreateReadTxn()
    it = txn.GetVertexIterator()
    nv = 0
    while it.IsValid() and nv < n:
        nv = nv + 1
        it.Next()
    return (True, str(nv))
)";
        code_write = R"(
from liblgraph_python_api import *
def Process(db, input):
    db.AddVertexLabel("v", [FieldSpec("id",FieldType.INT32,False),FieldSpec("name",FieldType.STRING,True)])
    txn = db.CreateWriteTxn()
    id = txn.AddVertex("v", {"id":1, "name":"001"})
    txn.Commit()