Fix unit test for visualization usages

Author: Julian Blank

README.rst

@@ -40,6 +40,7 @@ Documentation_ / Paper_ / Installation_ / Usage_ / Citation_ / Contact_
 
 
 .. _Installation:
+
 Installation
 ====================================================================
 
@@ -71,6 +72,7 @@ directory because otherwise not the in site-packages installed version will be u
 
 
 .. _Usage:
+
 Usage
 ==================================
 
@@ -109,6 +111,7 @@ A representative run of NSGA2 looks as follows:
 
 
 .. _Citation:
+
 Citation
 ====================================================================
 
@@ -129,6 +132,7 @@ Meanwhile, if you have used our framework for research purposes, please cite us
 
 
 .. _Contact:
+
 Contact
 ====================================================================
 

setup.py

@@ -63,6 +63,7 @@ def packages():
 
 
 data['long_description'] = readme()
+data['long_description_content_type'] = 'text/x-rst'
 data['packages'] = packages()
 
 

tests/visualization/test_visualization.py

@@ -9,7 +9,7 @@ class VisualizationTest(unittest.TestCase):
 
     def test_usages(self):
         folder = os.path.join(get_pymoo(), "pymoo", "usage", "visualization")
-        test_usage([os.path.join(folder, fname) for fname in os.listdir(folder)])
+        test_usage([os.path.join(folder, fname) for fname in os.listdir(folder) if fname.endswith("py")])
 
 if __name__ == '__main__':
     unittest.main()