lib/SimControl/environments/python.rb in SimControl-0.1.1 vs lib/SimControl/environments/python.rb in SimControl-0.1.2

@@ -1,17 +1,34 @@
+require "open3"
+
 module SimControl
   class PythonEnvironment < BaseEnvironment
     def initialize(script, args = {})
       @script = script
       @interpreter = args.delete(:interpreter)
       @virtualenv = args.delete(:virtualenv)
       raise "passing a virtualenv requires an interpreter" if @virtualenv and not @interpreter
     end
 
+    def script
+      @script
+    end
+
+    def args(scenario)
+      scenario.map { |k, v| "--#{ k } #{ v }" }.join " "
+    end
+
     def execute(scenario)
-      args = scenario.map { |k, v| "--#{ k } #{ v }" }.join " "
-      command = [interpreter, @script, args].reject(&:nil?).reject(&:empty?).join " "
-      `#{ command }`
+      stdin, stout, stderr, thread = Open3.popen3 command(scenario), chdir: basedir
+      thread.join
+    end
+
+    def basedir
+      File.dirname @script
+    end
+
+    def command(scenario)
+      [interpreter, script, args(scenario)].reject(&:nil?).reject(&:empty?).join " "
     end
 
     def interpreter
       if @virtualenv
         File.join(@virtualenv, "bin", @interpreter)