enable travis

ndanielsen · Jan 25, 2019 · 59f1014 · 59f1014
1 parent d8a2093
commit 59f1014
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -0,0 +1,9 @@
+language: python
+python:
+  - "3.3"
+# command to install dependencies
+install:
+  - pip install -r requirements.txt
+# command to run tests
+script:
+  - python tests/test.py
diff --git a/tests/test.py b/tests/test.py
@@ -0,0 +1,11 @@
+from clustering.equal_groups import EqualGroupsKMeans
+
+import numpy as np
+
+X = np.array([[1, 2], [4, 4], [1, 0], [3, 2], [4, 4], [4, 0]])
+
+clf = EqualGroupsKMeans(n_clusters=2)
+
+clf.fit(X)
+
+print (clf.labels_)