diff --git a/Dockerfile b/Dockerfile index 8e0430aa..82ce7ae5 100644 --- 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 new file mode 100644 index 00000000..fc29c28a --- /dev/null +++ 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) +