From 2ab70803ec47901ff9fab5a0961d3d023b6a7c62 Mon Sep 17 00:00:00 2001
From: jajupmochi <jajupmochi@gmail.com>
Date: Mon, 2 Mar 2020 15:47:57 +0100
Subject: [PATCH] fix bugs for pytest.

---
 .gitignore                         |  2 ++
 .travis.yml                        |  4 ++--
 gklearn/tests/__init__.py          |  8 ++++++++
 gklearn/tests/test_graphkernels.py | 32 ++++++++++++++++++++++++++++++++
 gklearn/tests/test_tools.py        |  1 +
 5 files changed, 45 insertions(+), 2 deletions(-)
 create mode 100644 gklearn/tests/test_graphkernels.py
 create mode 100644 gklearn/tests/test_tools.py

diff --git a/.gitignore b/.gitignore
index 86a5486..852f2a5 100644
--- a/.gitignore
+++ b/.gitignore
@@ -38,3 +38,5 @@ docs/log*
 *.egg-info
 dist/
 build/
+
+.coverage
diff --git a/.travis.yml b/.travis.yml
index 8f3f3b0..f8f21e0 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -16,10 +16,10 @@ before_install:
 install:
 - pip install -r requirements.txt
 - pip install wheel
-- python setup.py bdist_wheel
 
 script:
-- pytest --cov-report term --cov=gklearn gklearn/tests/
+- python setup.py bdist_wheel
+- pytest -v --cov-report term --cov=gklearn gklearn/tests/
 
 after_success:
 - codecov
diff --git a/gklearn/tests/__init__.py b/gklearn/tests/__init__.py
index e69de29..be702d9 100644
--- a/gklearn/tests/__init__.py
+++ b/gklearn/tests/__init__.py
@@ -0,0 +1,8 @@
+# -*-coding:utf-8 -*-
+"""tests - tests module.
+"""
+
+# info
+__version__ = "0.1"
+__author__ = "Linlin Jia"
+__date__ = "March 2020"
diff --git a/gklearn/tests/test_graphkernels.py b/gklearn/tests/test_graphkernels.py
new file mode 100644
index 0000000..17cd857
--- /dev/null
+++ b/gklearn/tests/test_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/gklearn/tests/test_tools.py b/gklearn/tests/test_tools.py
new file mode 100644
index 0000000..18778d4
--- /dev/null
+++ b/gklearn/tests/test_tools.py
@@ -0,0 +1 @@
+print('Hello graphkit-learn!')