Add similarity unit tests

Author: danieldk

.travis.yml

@@ -24,20 +24,22 @@ install:
     if [ "$TRAVIS_OS_NAME" == "osx" ]; then
       python3 -m venv venv
       source venv/bin/activate
-      pip install cffi virtualenv
+      pip install cffi virtualenv pytest
     fi
   - |
     if [ "$TRAVIS_OS_NAME" == "linux" ]; then
       python3.6 -m venv venv
       source venv/bin/activate
-      pip install cffi virtualenv
+      pip install cffi virtualenv pytest
     fi
   - cargo install pyo3-pack --vers 0.5.0
   - rustup default nightly-2019-02-07
   - rustup component add rustfmt
 
 script:
   - cargo fmt --all -- --check
+  - pyo3-pack develop
+  - pytest
   - pyo3-pack build --release
 deploy:
   provider: releases

tests/similarity.fifu

@@ -0,0 +1,68 @@
+import finalfusion
+
+SIMILARITY_ORDER_STUTTGART_10 = [
+        "Karlsruhe",
+        "Mannheim",
+        "München",
+        "Darmstadt",
+        "Heidelberg",
+        "Wiesbaden",
+        "Kassel",
+        "Düsseldorf",
+        "Leipzig",
+        "Berlin",
+    ];
+
+
+SIMILARITY_ORDER = [
+        "Potsdam",
+        "Hamburg",
+        "Leipzig",
+        "Dresden",
+        "München",
+        "Düsseldorf",
+        "Bonn",
+        "Stuttgart",
+        "Weimar",
+        "Berlin-Charlottenburg",
+        "Rostock",
+        "Karlsruhe",
+        "Chemnitz",
+        "Breslau",
+        "Wiesbaden",
+        "Hannover",
+        "Mannheim",
+        "Kassel",
+        "Köln",
+        "Danzig",
+        "Erfurt",
+        "Dessau",
+        "Bremen",
+        "Charlottenburg",
+        "Magdeburg",
+        "Neuruppin",
+        "Darmstadt",
+        "Jena",
+        "Wien",
+        "Heidelberg",
+        "Dortmund",
+        "Stettin",
+        "Schwerin",
+        "Neubrandenburg",
+        "Greifswald",
+        "Göttingen",
+        "Braunschweig",
+        "Berliner",
+        "Warschau",
+        "Berlin-Spandau",
+    ];
+
+def test_similarity_berlin_40():
+    embeds = finalfusion.Embeddings('tests/similarity.fifu')
+    for idx, sim in enumerate(embeds.similarity("Berlin", 40)):
+        assert SIMILARITY_ORDER[idx] == sim.word
+
+def test_similarity_stuttgart_10():
+    embeds = finalfusion.Embeddings('tests/similarity.fifu')
+    for idx, sim in enumerate(embeds.similarity("Stuttgart", 10)):
+        assert SIMILARITY_ORDER_STUTTGART_10[idx] == sim.word