Browse Source

fix bugs for pytest.

v0.1
jajupmochi 5 years ago
parent
commit
2ab70803ec
5 changed files with 45 additions and 2 deletions
  1. +2
    -0
      .gitignore
  2. +2
    -2
      .travis.yml
  3. +8
    -0
      gklearn/tests/__init__.py
  4. +32
    -0
      gklearn/tests/test_graphkernels.py
  5. +1
    -0
      gklearn/tests/test_tools.py

+ 2
- 0
.gitignore View File

@@ -38,3 +38,5 @@ docs/log*
*.egg-info
dist/
build/

.coverage

+ 2
- 2
.travis.yml View File

@@ -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

+ 8
- 0
gklearn/tests/__init__.py View File

@@ -0,0 +1,8 @@
# -*-coding:utf-8 -*-
"""tests - tests module.
"""

# info
__version__ = "0.1"
__author__ = "Linlin Jia"
__date__ = "March 2020"

+ 32
- 0
gklearn/tests/test_graphkernels.py View File

@@ -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()

+ 1
- 0
gklearn/tests/test_tools.py View File

@@ -0,0 +1 @@
print('Hello graphkit-learn!')

Loading…
Cancel
Save