run the right test suite by default

1 files changed, 16 insertions, 4 deletions

diff --git a/conftest.py b/conftest.py
index e0b4426..55e27a2 100644
--- a/conftest.py
+++ b/conftest.py
@@ -1,6 +1,18 @@
+import sys
+from _pytest.doctest import DoctestModule
+from py.path import local
+
+
+PY2 = sys.version_info[0] < 3
+
+
 collect_ignore = ['tests/messages/data']
 
-def pytest_configure(config):
-    import sys
-    if sys.version_info[0] < 3:
-        config.option.doctestmodules = True
+
+def pytest_collect_file(path, parent):
+    babel_path = local(__file__).dirpath().join('babel')
+    config = parent.config
+    if PY2:
+        if babel_path.common(path) == babel_path:
+            if path.ext == ".py":
+                return DoctestModule(path, parent)