Travis

.coveragerc

@@ -0,0 +1,10 @@
+[run]
+source=karatecluv
+[report]
+exclude_lines =
+    pragma: no cover
+    raise
+    except
+    register_parameter
+    warn
+    pass

.travis.yml

@@ -0,0 +1,27 @@
+language: shell
+
+os:
+  - linux
+  - osx
+  - windows
+
+env:
+  jobs:
+    - PYTHON_VERSION=3.6
+
+install:
+  - source script/conda.sh
+  - conda create --yes -n test python="${PYTHON_VERSION}"
+  - source activate test
+  - pip install codecov
+  - python setup.py install
+
+script:
+  - python setup.py test
+
+after_success:
+  - codecov
+
+notifications:
+  email: false
+

script/conda.sh

@@ -0,0 +1,26 @@
+#!/bin/bash
+
+if [ "${TRAVIS_OS_NAME}" = "linux" ]; then
+  wget -nv https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O miniconda.sh
+  chmod +x miniconda.sh
+  ./miniconda.sh -b
+  PATH=/home/travis/miniconda3/bin:${PATH}
+fi
+
+if [ "${TRAVIS_OS_NAME}" = "osx" ]; then
+  wget -nv https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-x86_64.sh -O miniconda.sh
+  chmod +x miniconda.sh
+  ./miniconda.sh -b
+  PATH=/Users/travis/miniconda3/bin:${PATH}
+fi
+
+
+if [ "${TRAVIS_OS_NAME}" = "windows" ]; then
+  choco install openssl.light
+  choco install miniconda3
+  PATH=/c/tools/miniconda3/Scripts:$PATH
+fi
+
+conda update --yes conda
+
+conda create --yes -n test python="${PYTHON_VERSION}"

setup.cfg

@@ -1,2 +1,11 @@
 [metadata]
 description-file = README.md
+
+[aliases]
+test=pytest
+
+[tool:pytest]
+addopts = --capture=no --cov
+
+
+

test/generic_test.py

@@ -0,0 +1,19 @@
+import networkx as nx
+
+from karateclub import DeepWalk
+
+def test_sampler():
+    """
+    Testing the sampler base class.
+    """
+    model = DeepWalk()
+
+    graph = nx.watts_strogatz_graph(200, 10, 0)
+
+    model.fit(graph)
+
+    embedding = model.get_embedding()
+
+    assert embedding.shape[1] = model.dimensions
+
+