From 3a85fb814b5d614f143d0564a76fe0e11f9ead2e Mon Sep 17 00:00:00 2001
From: yuvipanda <yuvipanda@gmail.com>
Date: Fri, 14 Dec 2018 13:29:14 -0800
Subject: [PATCH] Use pytest's capsys to test for stdout values

Works better with pytest's output capturing system
---
 tests/test_args.py | 13 +++++--------
 1 file changed, 5 insertions(+), 8 deletions(-)

diff --git a/tests/test_args.py b/tests/test_args.py
index d4c04d84..2fba2027 100644
--- a/tests/test_args.py
+++ b/tests/test_args.py
@@ -6,18 +6,15 @@ import pytest
 import logging
 from repo2docker.__main__ import make_r2d
 from repo2docker import __version__
-from contextlib import redirect_stdout
-import io
 
-def test_version():
+
+def test_version(capsys):
     """
     Test passing '--version' to repo2docker
     """
-    stdout = io.StringIO()
-    with redirect_stdout(stdout):
-        with pytest.raises(SystemExit):
-            make_r2d(['--version'])
-    assert stdout.getvalue().strip() == __version__
+    with pytest.raises(SystemExit):
+        make_r2d(['--version'])
+    assert capsys.readouterr().out == f"{__version__}\n"
 
 def test_simple():
     """