Skip to content

Commit cfb4b95

Browse files
authored
Merge pull request #661 from Kaggle/add-annoy-tests
Add test for annoy package
2 parents 448196c + cd5d279 commit cfb4b95

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

tests/test_annoy.py

+11
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
import unittest
2+
3+
from annoy import AnnoyIndex
4+
5+
6+
class TestAnnoy(unittest.TestCase):
7+
def test_tree(self):
8+
t = AnnoyIndex(5, 'angular')
9+
t.add_item(1, [1,2,3,4,5])
10+
11+
self.assertTrue(t.build(1))

0 commit comments

Comments
 (0)