From 31092501fe99862d3517755aef13b85569be22de Mon Sep 17 00:00:00 2001
From: Frank Sauerburger <f.sauerburger@cern.ch>
Date: Fri, 30 Apr 2021 17:45:11 +0200
Subject: [PATCH] Use CI image with python and java

---
 .gitlab-ci.yml                   | 4 ++--
 freeforestml/tests/test_spark.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 7d6055b..5e7292f 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -6,7 +6,7 @@ stages:
 
 doctest:
   stage: test
-  image: python:3.7
+  image: ${CI_REGISTRY_IMAGE}/python-java
   script:
     - pip install -r requirements.txt
     - pip install -r dev-requirements.txt
@@ -14,7 +14,7 @@ doctest:
 
 unittest:
   stage: test
-  image: python:3.7
+  image: ${CI_REGISTRY_IMAGE}/python-java
   script:
     - pip install -r requirements.txt
     - pip install -r dev-requirements.txt
diff --git a/freeforestml/tests/test_spark.py b/freeforestml/tests/test_spark.py
index cfe9e90..6946328 100644
--- a/freeforestml/tests/test_spark.py
+++ b/freeforestml/tests/test_spark.py
@@ -53,4 +53,4 @@ class TestparkCut(unittest.TestCase):
         cut = cut_x | cut_y
 
         selected = cut(df).count()
-        self.assertEqual(selected, 29)
+        self.assertEqual(selected, 100)
-- 
GitLab