Index: tests/test_training.py =================================================================== diff -u --- tests/test_training.py (revision 0) +++ tests/test_training.py (revision 87891e5102d69f7eb8e771f3bb91a9cf72e3b83f) @@ -0,0 +1,41 @@ +########################################################################### +# +# Copyright (c) 2020-2024 Diality Inc. - All Rights Reserved. +# +# THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM, IN PART OR IN +# WHOLE, WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER. +# +# @file test_training.py +# +# @author (last) Sean Nash +# @date (last) 02-Oct-2025 +# @author (original) Sean Nash +# @date (original) 02-Oct-2025 +# +############################################################################ + +import sys +sys.path.append("../leahi_dialin") +from time import sleep +from leahi_dialin.td.treatment_delivery import TD +from leahi_dialin.td.modules.constants import RESET, NO_RESET + +if __name__ == "__main__": + # create an TD object called td + td = TD(log_level="DEBUG") + sleep(2) + + # log in to TD as tester + if td.cmd_log_in_to_td() == 0: + exit(1) + sleep(3.5) + + # override training value + td.cmd_training_test_override(10, NO_RESET) + sleep(5) + + # reset override of training value + td.cmd_training_test_override(0, RESET) + sleep(1) + + exit(0)