Add timeout for PyLint and PyUnit Adapter

lib/py_lint_adapter.rb

@@ -11,11 +11,17 @@ class PyLintAdapter < TestingFrameworkAdapter
     count = captures.second
     passed = captures.first
     failed = count - passed
+    begin
+      Timeout.timeout(2.seconds) do
         assertion_error_matches = output[:stdout].scan(ASSERTION_ERROR_REGEXP).map { |match|
           test = match.first.strip
           description = match.second.strip
           "#{test}: #{description}"
         }.flatten || []
+      end
+    rescue Timeout::Error
+      assertion_error_matches = []
+    end
     {count: count, failed: failed, error_messages: assertion_error_matches}
   end
 end

lib/py_unit_adapter.rb

@@ -14,6 +14,8 @@ class PyUnitAdapter < TestingFrameworkAdapter
     failed = failures_matches ? failures_matches.captures.try(:first).to_i : 0
     error_matches = ERRORS_REGEXP.match(output[:stderr])
     errors = error_matches ? error_matches.captures.try(:first).to_i : 0
+    begin
+      Timeout.timeout(2.seconds) do
         assertion_error_matches = output[:stderr].scan(ASSERTION_ERROR_REGEXP).map { |match|
           testname = match[1]
           error = match[3].strip
@@ -24,6 +26,10 @@ class PyUnitAdapter < TestingFrameworkAdapter
             "#{testname}: #{error}"
           end
         }.flatten || []
+      end
+    rescue Timeout::Error
+      assertion_error_matches = []
+    end
     {count: count, failed: failed + errors, error_messages: assertion_error_matches}
   end
 end