Add more unit tests (#8)

WillianFuks · Jun 9, 2021 · e0515d8 · e0515d8
1 parent 0aa8a73
commit e0515d8
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/tests/fixtures/all_clicks_set/judgments.gz b/tests/fixtures/all_clicks_set/judgments.gz
diff --git a/tests/test_cy_DBN.pyx b/tests/test_cy_DBN.pyx
@@ -1896,15 +1896,23 @@ cdef bint test_not_null_converence():
     cdef:
         DBNModel model = DBN()
 
-    model.fit('tests/fixtures/null_test', iters=1)
+    model.fit('tests/fixtures/null_test', iters=10)
     return True
 
 
 cdef bint test_long_list_null_converence():
     cdef:
         DBNModel model = DBN()
 
-    model.fit('tests/fixtures/eighty_skus', iters=2)
+    model.fit('tests/fixtures/eighty_skus', iters=10)
+    return True
+
+
+cdef bint test_all_clicks_set():
+    cdef:
+        DBNModel model = DBN()
+
+    model.fit('tests/fixtures/all_clicks_set', iters=10)
     return True
 
 
@@ -1930,6 +1938,7 @@ cpdef run_tests():
     assert test_export_judgments()
     assert test_not_null_converence()
     assert test_long_list_null_converence()
+    assert test_all_clicks_set()
 
 
 if __name__ == '__main__':