From 562ddf775e6bf0a2d383e326709637dd665a85a1 Mon Sep 17 00:00:00 2001
From: jajupmochi <jajupmochi@gmail.com>
Date: Mon, 2 Mar 2020 15:07:34 +0100
Subject: [PATCH] add tests.

---
 .travis.yml                             | 12 ++++++++++--
 gklearn/tests/__init__.py               |  0
 gklearn/tests/requirements.txt          |  7 +++++++
 gklearn/tests/tests_graphkernels.py     | 32 ++++++++++++++++++++++++++++++++
 {tests => gklearn/tests}/tests_tools.py |  0
 5 files changed, 49 insertions(+), 2 deletions(-)
 create mode 100644 gklearn/tests/__init__.py
 create mode 100644 gklearn/tests/requirements.txt
 create mode 100644 gklearn/tests/tests_graphkernels.py
 rename {tests => gklearn/tests}/tests_tools.py (100%)

diff --git a/.travis.yml b/.travis.yml
index 747f953..6ba124f 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -1,17 +1,25 @@
 language: python
 python:
 - '3.6.5'
+- '3.5'
+- '3.6'
+- '3.7'
+- '3.8'
 before_install:
+- python --version
+- pip install -U pip
+- pip install -U pytest
 - pip install codecov
 - pip install coverage
+- pip install pytest-cov
 
 install:
 - pip install -r requirements.txt
 - pip install wheel
+- python setup.py bdist_wheel
 
 script:
-- python setup.py bdist_wheel
-- coverage run tests/tests_tools.py
+- pytest --cov-report term --cov=investpy gklearn/tests/
 
 after_success:
 - codecov
diff --git a/gklearn/tests/__init__.py b/gklearn/tests/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/gklearn/tests/requirements.txt b/gklearn/tests/requirements.txt
new file mode 100644
index 0000000..a48620b
--- /dev/null
+++ b/gklearn/tests/requirements.txt
@@ -0,0 +1,7 @@
+numpy==1.15.2
+scipy==1.1.0
+matplotlib==3.0.0
+networkx==2.2
+scikit-learn==0.20.0
+tabulate==0.8.2
+tqdm==4.26.0
diff --git a/gklearn/tests/tests_graphkernels.py b/gklearn/tests/tests_graphkernels.py
new file mode 100644
index 0000000..7154972
--- /dev/null
+++ b/gklearn/tests/tests_graphkernels.py
@@ -0,0 +1,32 @@
+"""Tests of graph kernels.
+"""
+
+#import pytest
+from gklearn.utils.graphfiles import loadDataset
+
+
+def test_spkernel():
+    """Test shortest path kernel.
+    """
+    from gklearn.kernels.spKernel import spkernel
+    from gklearn.utils.kernels import deltakernel, gaussiankernel, kernelproduct
+    import functools
+    import multiprocessing
+
+    ds_file = '../../datasets/Alkane/dataset.ds'
+    ds_y = '../../datasets/Alkane/dataset_boiling_point_names.txt'
+    Gn, y = loadDataset(ds_file, filename_y=ds_y)
+    Gn = Gn[0:10]
+    y = y[0:10]
+    
+    mixkernel = functools.partial(kernelproduct, deltakernel, gaussiankernel)
+    try:
+        Kmatrix, run_time, idx = spkernel(Gn, node_label=None, node_kernels=
+            {'symb': deltakernel, 'nsymb': gaussiankernel, 'mix': mixkernel},
+            n_jobs=multiprocessing.cpu_count(), verbose=True)
+    except Exception as exception:
+        assert False, exception
+    
+
+if __name__ == "__main__":
+    test_spkernel()
\ No newline at end of file
diff --git a/tests/tests_tools.py b/gklearn/tests/tests_tools.py
similarity index 100%
rename from tests/tests_tools.py
rename to gklearn/tests/tests_tools.py