From 4dd26b61165654e7fce7dfbd9e6e3f94fbb5e696 Mon Sep 17 00:00:00 2001 From: Jacob Silterra Date: Wed, 29 Nov 2023 15:04:59 -0500 Subject: [PATCH] Add a very simple unit test --- tests/test_create_sybilnet.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 tests/test_create_sybilnet.py diff --git a/tests/test_create_sybilnet.py b/tests/test_create_sybilnet.py new file mode 100644 index 0000000..9574c5a --- /dev/null +++ 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