Browse Source

add tests.

v0.1
jajupmochi 5 years ago
parent
commit
562ddf775e
5 changed files with 49 additions and 2 deletions
  1. +10
    -2
      .travis.yml
  2. +0
    -0
      gklearn/tests/__init__.py
  3. +7
    -0
      gklearn/tests/requirements.txt
  4. +32
    -0
      gklearn/tests/tests_graphkernels.py
  5. +0
    -0
      gklearn/tests/tests_tools.py

+ 10
- 2
.travis.yml View File

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

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


+ 7
- 0
gklearn/tests/requirements.txt View File

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

+ 32
- 0
gklearn/tests/tests_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()

tests/tests_tools.py → gklearn/tests/tests_tools.py View File


Loading…
Cancel
Save