test: also look for examples in [.../dist-packages]/migen/examples/.

m-labs · Dec 1, 2015 · 202e23b · 202e23b
1 parent 2bf1110
commit 202e23b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/migen/test/test_examples.py b/migen/test/test_examples.py
@@ -8,7 +8,11 @@ def _make_test_method(name, foldername):
     def test_method(self):
         filename = name + ".py"
         example_path = os.path.abspath(
-            os.path.join(os.path.dirname(__file__), "..", "..", "examples"))
+            os.path.join(os.path.dirname(__file__), "..", "examples"))
+        if not os.path.isfile(example_path):
+            # we're running from the source tree
+            example_path = os.path.abspath(
+                os.path.join(os.path.dirname(__file__), "..", "..", "examples"))
         filepath = os.path.join(example_path, foldername, filename)
         subprocess.check_call(
             [sys.executable, filepath],