Kaggle · andrewekhalel · Nov 6, 2018
diff --git a/Dockerfile b/Dockerfile
@@ -475,6 +475,7 @@ RUN pip install flashtext && \
    pip install PDPbox && \
    pip install ggplot && \
    pip install cesium && \
+    pip install edafa && \
    ##### ^^^^ Add new contributions above here ^^^^ #####
    # clean up pip cache
    rm -rf /root/.cache/pip/*

diff --git a/tests/test_edafa.py b/tests/test_edafa.py
@@ -0,0 +1,21 @@
+import unittest
+
+from edafa import ClassPredictor
+
+class myPredictor(ClassPredictor):
+	def __init__(self,*args,**kwargs):
+		super().__init__(*args,**kwargs)
+
+	def predict_patches(self,patches):
+		return 0
+
+class TestEdafa(unittest.TestCase):
+	def test_init(self):
+		conf = '{"augs":["NO",\
+		"FLIP_UD",\
+		"FLIP_LR"],\
+		"mean":"ARITH"}'
+		p = myPredictor(conf)
+		pred = p.predict_patches(None)
+		self.assertEqual(0, pred)
+