Run pytest in julia process

Author: tkf

src/convenience.jl

@@ -145,8 +145,29 @@ function install_dependency(package; force=false, dry_run=false)
 end
 
 
-function test_replhelper()
-    command = `$(PyCall.pyprogramname) -m pytest`
+function test_replhelper(args=``; inprocess=false, kwargs...)
+    if inprocess
+        test_replhelper_inprocess(args; kwargs...)
+    else
+        test_replhelper_cli(args)
+    end
+end
+
+function test_replhelper_inprocess(args; revise=true)
+    IPython._start_ipython(:ipython_options)  # setup replhelper.core._Main
+    if revise
+        pyimport("replhelper")[:revise]()
+    end
+    cd(@__DIR__) do
+        code = pyimport("pytest")[:main](collect(args))
+        if code != 0
+            error("$(`pytest $args`) failed with code $code")
+        end
+    end
+end
+
+function test_replhelper_cli(args)
+    command = `$(PyCall.pyprogramname) -m pytest $args`
     @info command
     cd(@__DIR__) do
         run(command)

src/replhelper/core.py

@@ -3,8 +3,9 @@
 
 from __future__ import print_function
 
-import warnings
 import sys
+import types
+import warnings
 
 try:
     from importlib import reload
@@ -147,6 +148,25 @@ def wrapped(*args, **kwargs):
     return wrapped
 
 
+def get_api(main):
+    if main is None:
+        return None
+    return main._JuliaNameSpace__julia
+
+
+def get_main(**kwargs):
+    """
+    Create or get cached `Main`.
+
+    Caching is required to avoid re-writing to `_Main` when re-entering
+    to IPython session (where `user_ns` would be ignored).
+    """
+    global _Main
+    if _Main is None:
+        _Main = JuliaNameSpace(JuliaAPI(**kwargs))
+    return _Main
+
+
 def ipython_options(**kwargs):
     global _Main
     from traitlets.config import Config
@@ -198,12 +218,27 @@ def customized_ipython(**kwargs):
 
 def revise():
     """Ad-hoc hot reload."""
+
+    Main = _Main
+
     import replhelper
     reload(replhelper.core)
 
-    if _Main is not None:
-        _Main.__class__ = replhelper.core.JuliaNameSpace
-        _Main._JuliaNameSpace__julia.__class__ = replhelper.core.JuliaAPI
-        replhelper.core._Main = _Main
+    if Main is not None:
+        Main.__class__ = replhelper.core.JuliaNameSpace
+        Main._JuliaNameSpace__julia.__class__ = replhelper.core.JuliaAPI
+        replhelper.core._Main = Main
+
+    try:
+        replhelper.tests
+    except AttributeError:
+        return
+
+    # *Try* reloading modules `replhelper.tests.*`.  If there are
+    # dependencies between those modules, it's not going to work.
+    for (name, module) in sorted(vars(replhelper.tests).items(),
+                                 key=lambda pair: pair[0]):
+        if isinstance(module, types.ModuleType):
+            reload(module)
 
     reload(replhelper)

src/replhelper/tests/conftest.py

@@ -0,0 +1,18 @@
+import pytest
+
+from .. import core
+
+
+@pytest.fixture
+def Main():
+    """ pytest fixture for providing a Julia `Main` name space. """
+    if core._Main is None:
+        pytest.skip("Main not configured (not called from IPython.jl?)")
+    else:
+        return core._Main
+
+
+@pytest.fixture
+def julia(Main):
+    """ pytest fixture for providing a `JuliaAPI` instance. """
+    return core.get_api(Main)

test/runtests.jl

@@ -15,6 +15,6 @@ end
 include("test_julia_repl.jl")
 include("test_convenience.jl")
 
-IPython.test_replhelper()
+IPython.test_replhelper(inprocess=true)
 
 # end  # module