Browse Source

fix a bug for pytest.

v0.1
jajupmochi 5 years ago
parent
commit
1306032de3
2 changed files with 0 additions and 33 deletions
  1. +0
    -32
      gklearn/tests/tests_graphkernels.py
  2. +0
    -1
      gklearn/tests/tests_tools.py

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

@@ -1,32 +0,0 @@
"""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()

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

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

Loading…
Cancel
Save