###########################################################################
#
# 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)