From c6171a85e986a69c670bb03b19574992cd94cbc4 Mon Sep 17 00:00:00 2001
From: nvbn <nvbn.rm@gmail.com>
Date: Mon, 24 Aug 2015 11:25:49 +0300
Subject: [PATCH] Improve readme test

---
 tests/test_readme.py | 14 ++++++--------
 1 file changed, 6 insertions(+), 8 deletions(-)

diff --git a/tests/test_readme.py b/tests/test_readme.py
index 2cd5508b..715f3bc7 100644
--- a/tests/test_readme.py
+++ b/tests/test_readme.py
@@ -2,18 +2,16 @@ from pathlib import Path
 
 
 def test_readme():
-    with open('README.md') as f:
+    project_root = Path(__file__).parent.parent
+    with project_root.joinpath('README.md').open() as f:
         readme = f.read()
 
-        bundled = Path(__file__).parent.parent \
+        bundled = project_root \
             .joinpath('thefuck') \
             .joinpath('rules') \
             .glob('*.py')
 
         for rule in bundled:
-            if rule.stem != '__init__' and rule.stem not in readme:
-                raise Exception('Missing rule "{}" in README.md'.format(rule.stem))
-
-
-if __name__ == '__main__':
-    test_readme()
+            if rule.stem != '__init__':
+                assert rule.stem in readme,\
+                    'Missing rule "{}" in README.md'.format(rule.stem)