Some assertions were not being run. Move them so that they are run

Author: adamtheturtle

tests/test_find_extra_reqs.py

@@ -7,7 +7,7 @@
 import pytest
 import pretend
 
-from pip_check_reqs import find_extra_reqs, common
+from pip_check_reqs import find_extra_reqs, common, __version__
 
 
 @pytest.fixture
@@ -90,7 +90,8 @@ def test_main_failure(monkeypatch, caplog, fake_opts):
 
     with pytest.raises(SystemExit) as excinfo:
         find_extra_reqs.main()
-        assert excinfo.value == 1
+
+    assert excinfo.value.code == 1
 
     assert caplog.records[0].message == \
         'Extra requirements:'
@@ -108,7 +109,8 @@ def test_main_no_spec(monkeypatch, caplog, fake_opts):
 
     with pytest.raises(SystemExit) as excinfo:
         find_extra_reqs.main()
-        assert excinfo.value == 2
+
+    assert excinfo.value.code == 2
 
     assert fake_opts.error.calls
 
@@ -164,4 +166,5 @@ def test_main_version(monkeypatch, caplog, fake_opts):
 
     with pytest.raises(SystemExit) as excinfo:
         find_extra_reqs.main()
-        assert excinfo.value == 'version'
+
+    assert str(excinfo.value) == __version__