Add a very simple unit test

reginabarzilaygroup · Nov 29, 2023 · 4dd26b6 · 4dd26b6
1 parent c2c17ca
commit 4dd26b6
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/tests/test_create_sybilnet.py b/tests/test_create_sybilnet.py
@@ -0,0 +1,18 @@
+import argparse
+import datetime
+import os
+
+from sybil import Serie, Sybil
+
+def test_create_sybilnet():
+    from sybil.models.sybil import SybilNet
+
+    fake_args = argparse.Namespace(
+        dropout=0.1,
+        max_followup=5,
+        )
+
+    sybil_net = SybilNet(fake_args)
+
+    assert sybil_net.hidden_dim == 512
+    assert sybil_net.prob_of_failure_layer is not None