From 4018efc50d2a1ebafcf146079be815f3543334ed Mon Sep 17 00:00:00 2001
From: rdb <git@rdb.name>
Date: Tue, 19 Sep 2017 17:49:04 +0200
Subject: [PATCH] Fix travis build

---
 .travis.yml               |  1 +
 makepanda/test_imports.py | 12 +++++++++---
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/.travis.yml b/.travis.yml
index 7d8a3003b6..36e4749240 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -35,6 +35,7 @@ addons:
     - python-dev
     - python3-dev
     - zlib1g-dev
+    - fakeroot
 script: $PYTHONV makepanda/makepanda.py --everything --git-commit $TRAVIS_COMMIT $FLAGS --threads 4 && LD_LIBRARY_PATH=built/lib PYTHONPATH=built $PYTHONV makepanda/test_imports.py
 notifications:
   irc:
diff --git a/makepanda/test_imports.py b/makepanda/test_imports.py
index b5ab5c1977..290ec03e13 100644
--- a/makepanda/test_imports.py
+++ b/makepanda/test_imports.py
@@ -7,13 +7,19 @@ import os, importlib
 import direct.showbase.VerboseImport
 
 
+import imp
 import panda3d
 dir = os.path.dirname(panda3d.__file__)
 
-for basename in os.listdir(dir):
-    module, ext = os.path.splitext(basename)
+extensions = set()
+for suffix in imp.get_suffixes():
+    extensions.add(suffix[0])
 
-    if ext in ('.pyd', '.so'):
+for basename in os.listdir(dir):
+    module = basename.split('.', 1)[0]
+    ext = basename[len(module):]
+
+    if ext in extensions:
         importlib.import_module('panda3d.%s' % (module))