Fix unit test for visualization usages

ahmmedshakil · Apr 3, 2020 · 3e822b1 · 3e822b1
1 parent 28e9cdb
commit 3e822b1
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/README.rst b/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
 ====================================================================
 

diff --git a/setup.py b/setup.py
@@ -63,6 +63,7 @@ def packages():
 
 
 data['long_description'] = readme()
+data['long_description_content_type'] = 'text/x-rst'
 data['packages'] = packages()
 
 

diff --git a/tests/visualization/test_visualization.py b/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()
Original file line number	Diff line number	Diff line change
Expand Up		@@ -63,6 +63,7 @@ def packages():


		data['long_description'] = readme()
		data['long_description_content_type'] = 'text/x-rst'
		data['packages'] = packages()


Expand Down