Add pytest back in

Add pytest test

Author: SohierDane

Dockerfile.tmpl

@@ -320,7 +320,7 @@ RUN pip install mpld3 \
         python-louvain \
         pyexcel-ods \
         sklearn-pandas \
-        stemming \    
+        stemming \
         # b/266272046 prophet 1.1.2 breaks the test
         prophet==1.1.1 \
         # b/283847935 holidays 0.25 is broken
@@ -342,7 +342,8 @@ RUN pip install mpld3 \
         geoplot \
         eli5 \
         kaggle \
-        mock && \
+        mock \
+        pytest && \
     /tmp/clean-layer.sh
 
 RUN rm -rf /opt/conda/lib/python3.10/site-packages/numpy-1.23.5.dist-info*
@@ -361,9 +362,6 @@ RUN pip install tensorpack && \
         memory_profiler && \
     /tmp/clean-layer.sh
 
-
-
-
 # install cython & cysignals before pyfasttext
 RUN pip install cython \
         cysignals \

tests/test_pytest.py

@@ -0,0 +1,10 @@
+
+import subprocess
+import unittest
+
+
+class test_pytest(unittest.TestCase):
+    def test_pytest_on_pandas_test(self):
+        cmd = 'pytest /input/tests/test_pandas.py'
+        result = subprocess.run(cmd, shell=True, capture_output=True)
+        self.assertEqual(result.returncode, 0)