--- mayavi2-3.1.0.orig/setup.py
+++ mayavi2-3.1.0/setup.py
@@ -335,6 +335,8 @@
     config.add_subpackage('enthought.tvtk')
     config.add_subpackage('enthought')
     config.add_data_dir('enthought/mayavi/core/lut')
+    config.add_data_dir('enthought/mayavi/tests/data')
+    config.add_data_dir('enthought/mayavi/tests/csv_files')
 
     # Image files.
     for root, dirs, files in os.walk('enthought'):