MTPPy2.0 0.1.5__py3-none-any.whl

mtppy/part3/data_assembly_set/active_elements/drives/bin_drv_test.py

@@ -0,0 +1,503 @@
+"""Test Suite for BinDrv."""
+
+__copyright__ = (
+    "Copyright (c) 2026 Dresden University of Technology, Process-to-Order Group"
+)
+__license__ = "MIT"
+import pytest
+
+from mtppy.utils.testing import data_object_set_modes
+from mtppy.part3.data_assembly_set.active_elements.drives.bin_drv import BinDrv
+
+
+def init_bin_drv(
+    init_op_mode="off",
+    init_src_mode="int",
+    rev_fbk_calc=True,
+    fwd_fbk_calc=True,
+    safe_pos=0,
+    fwd_en=True,
+    rev_en=False,
+    perm_en=False,
+    intl_en=False,
+    prot_en=False,
+):
+    instance = BinDrv(
+        "tag",
+        tag_description="",
+        rev_fbk_calc=rev_fbk_calc,
+        fwd_fbk_calc=fwd_fbk_calc,
+        safe_pos=safe_pos,
+        fwd_en=fwd_en,
+        rev_en=rev_en,
+        perm_en=perm_en,
+        intl_en=intl_en,
+        prot_en=prot_en,
+    )
+    data_object_set_modes(instance, init_op_mode, init_src_mode)
+    return instance
+
+
+test_scenario = [
+    ("off", "int", "set_fwd_op", False),
+    ("off", "man", "set_fwd_op", False),
+    ("op", "int", "set_fwd_op", True),
+    ("op", "man", "set_fwd_op", True),
+    ("aut", "int", "set_fwd_op", False),
+    ("aut", "man", "set_fwd_op", False),
+    ("off", "int", "set_fwd_aut", False),
+    ("off", "man", "set_fwd_aut", False),
+    ("op", "int", "set_fwd_aut", False),
+    ("op", "man", "set_fwd_aut", False),
+    ("aut", "int", "set_fwd_aut", True),
+    ("aut", "man", "set_fwd_aut", True),
+    ("off", "int", "set_rev_op", False),
+    ("off", "man", "set_rev_op", False),
+    ("op", "int", "set_rev_op", True),
+    ("op", "man", "set_rev_op", True),
+    ("aut", "int", "set_rev_op", False),
+    ("aut", "man", "set_rev_op", False),
+    ("off", "int", "set_rev_aut", False),
+    ("off", "man", "set_rev_aut", False),
+    ("op", "int", "set_rev_aut", False),
+    ("op", "man", "set_rev_aut", False),
+    ("aut", "int", "set_rev_aut", True),
+    ("aut", "man", "set_rev_aut", True),
+]
+
+
+def test_fwd_rev():
+    for op_mode, src_mode, set_command, expected_result in test_scenario:
+        for command in [True, False]:
+            bin_drv = init_bin_drv(
+                init_op_mode=op_mode, init_src_mode=src_mode, fwd_en=True, rev_en=True
+            )
+            eval(f"bin_drv.{set_command}({command})")
+            print(
+                f"Scenario: mode {op_mode} {src_mode}, {set_command}, expected: {expected_result}"
+            )
+            if set_command in ["set_fwd_op", "set_fwd_aut"]:
+                if command:
+                    assert bin_drv["FwdCtrl"] == expected_result
+                    assert bin_drv.get_fwd_fbk() == expected_result
+                    assert bin_drv["RevCtrl"] == False
+                    assert bin_drv.get_rev_fbk() == False
+                else:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv.get_fwd_fbk() == False
+                    assert bin_drv["RevCtrl"] == False
+                    assert bin_drv.get_rev_fbk() == False
+            elif set_command in ["set_rev_op", "set_rev_aut"]:
+                if command:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv.get_fwd_fbk() == False
+                    assert bin_drv["RevCtrl"] == expected_result
+                    assert bin_drv.get_rev_fbk() == expected_result
+                else:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv.get_fwd_fbk() == False
+                    assert bin_drv["RevCtrl"] == False
+                    assert bin_drv.get_rev_fbk() == False
+
+
+def test_fwd_rev_en():
+    for op_mode, src_mode, set_command, _ in test_scenario:
+        for command in [True, False]:
+            bin_drv = init_bin_drv(
+                init_op_mode=op_mode, init_src_mode=src_mode, fwd_en=False, rev_en=False
+            )
+            eval(f"bin_drv.{set_command}({command})")
+            print(
+                f"Scenario: mode {op_mode} {src_mode}, {set_command}, expected: False"
+            )
+            assert bin_drv["FwdCtrl"] == False
+            assert bin_drv["RevCtrl"] == False
+
+
+def test_fwd_rev_trip():
+    for op_mode, src_mode, set_command, _ in test_scenario:
+        for command in [True, False]:
+            bin_drv = init_bin_drv(
+                init_op_mode=op_mode, init_src_mode=src_mode, fwd_en=True, rev_en=True
+            )
+            bin_drv.set_trip(False)
+            eval(f"bin_drv.{set_command}({command})")
+            print(
+                f"Scenario: mode {op_mode} {src_mode}, {set_command}, expected: False"
+            )
+            assert bin_drv["SafePosAct"] == True
+            assert bin_drv["Trip"] == False
+            assert bin_drv["FwdCtrl"] == False
+            assert bin_drv["RevCtrl"] == False
+
+
+@pytest.mark.parametrize(
+    "op_mode, src_mode, set_command, expected_result", test_scenario
+)
+def test_fwd_rev_permit_en_true(op_mode, src_mode, set_command, expected_result):
+    for command in [True, False]:
+        bin_drv = init_bin_drv(
+            init_op_mode=op_mode,
+            init_src_mode=src_mode,
+            fwd_en=True,
+            rev_en=True,
+            perm_en=True,
+        )
+        bin_drv.set_permit(False)
+        eval(f"bin_drv.{set_command}({command})")
+        print(f"Scenario: mode {op_mode} {src_mode}, {set_command}, expected: False")
+        assert bin_drv["SafePosAct"] == True
+        assert bin_drv["Permit"] == False
+        assert bin_drv["FwdCtrl"] == False
+        assert bin_drv["RevCtrl"] == False
+        assert bin_drv.get_fwd_fbk() == False
+        assert bin_drv.get_rev_fbk() == False
+
+        for command_2 in [True, False]:
+            bin_drv = init_bin_drv(
+                init_op_mode=op_mode,
+                init_src_mode=src_mode,
+                fwd_en=True,
+                rev_en=True,
+                perm_en=True,
+            )
+            bin_drv.set_permit(True)
+            eval(f"bin_drv.{set_command}({command_2})")
+            print(
+                f"Scenario: mode {op_mode} {src_mode}, {set_command}, expected: {expected_result}"
+            )
+
+            assert bin_drv["Permit"] == True
+            assert bin_drv["SafePosAct"] == False
+            if set_command in ["set_fwd_op", "set_fwd_aut"]:
+                if command_2:
+                    assert bin_drv["FwdCtrl"] == expected_result
+                    assert bin_drv.get_fwd_fbk() == expected_result
+                    assert bin_drv["RevCtrl"] == False
+                    assert bin_drv.get_rev_fbk() == False
+                else:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv.get_fwd_fbk() == False
+                    assert bin_drv["RevCtrl"] == False
+                    assert bin_drv.get_rev_fbk() == False
+            elif set_command in ["set_rev_op", "set_rev_aut"]:
+                if command_2:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv.get_fwd_fbk() == False
+                    assert bin_drv["RevCtrl"] == expected_result
+                    assert bin_drv.get_rev_fbk() == expected_result
+                else:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv.get_fwd_fbk() == False
+                    assert bin_drv["RevCtrl"] == False
+                    assert bin_drv.get_rev_fbk() == False
+
+
+def test_fwd_rev_permit_en_false():
+    for op_mode, src_mode, set_command, expected_result in test_scenario:
+        for command in [True, False]:
+            ana_drv = init_bin_drv(
+                init_op_mode=op_mode,
+                init_src_mode=src_mode,
+                fwd_en=True,
+                rev_en=True,
+                perm_en=False,
+            )
+            ana_drv.set_permit(True)
+            eval(f"ana_drv.{set_command}({command})")
+            print(
+                f"Scenario: mode {op_mode} {src_mode}, {set_command}, expected: {expected_result}"
+            )
+
+            assert ana_drv["Permit"] == True
+            assert ana_drv["SafePosAct"] == False
+            if set_command in ["set_fwd_op", "set_fwd_aut"]:
+                if command:
+                    assert ana_drv["FwdCtrl"] == expected_result
+                    assert ana_drv["RevCtrl"] == False
+                else:
+                    assert ana_drv["FwdCtrl"] == False
+                    assert ana_drv["RevCtrl"] == False
+            elif set_command in ["set_rev_op", "set_rev_aut"]:
+                if command:
+                    assert ana_drv["FwdCtrl"] == False
+                    assert ana_drv["RevCtrl"] == expected_result
+                else:
+                    assert ana_drv["FwdCtrl"] == False
+                    assert ana_drv["RevCtrl"] == False
+
+
+def test_fwd_rev_interlock_en_true():
+    for op_mode, src_mode, set_command, _ in test_scenario:
+        for command in [True, False]:
+            bin_drv = init_bin_drv(
+                init_op_mode=op_mode,
+                init_src_mode=src_mode,
+                fwd_en=True,
+                rev_en=True,
+                intl_en=True,
+            )
+            bin_drv.set_interlock(False)
+            eval(f"bin_drv.{set_command}({command})")
+            print(
+                f"Scenario: mode {op_mode} {src_mode}, {set_command}, expected: False"
+            )
+
+            assert bin_drv["Interlock"] == False
+            assert bin_drv["SafePosAct"] == True
+            assert bin_drv["FwdCtrl"] == False
+            assert bin_drv["RevCtrl"] == False
+
+    for op_mode, src_mode, set_command, expected_result in test_scenario:
+        for command in [True, False]:
+            bin_drv = init_bin_drv(
+                init_op_mode=op_mode,
+                init_src_mode=src_mode,
+                fwd_en=True,
+                rev_en=True,
+                intl_en=True,
+            )
+            bin_drv.set_interlock(True)
+            eval(f"bin_drv.{set_command}({command})")
+            print(
+                f"Scenario: mode {op_mode} {src_mode}, {set_command}, expected: {expected_result}"
+            )
+
+            assert bin_drv["Interlock"] == True
+            assert bin_drv["SafePosAct"] == False
+            if set_command in ["set_fwd_op", "set_fwd_aut"]:
+                if command:
+                    assert bin_drv["FwdCtrl"] == expected_result
+                    assert bin_drv["RevCtrl"] == False
+                else:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv["RevCtrl"] == False
+            elif set_command in ["set_rev_op", "set_rev_aut"]:
+                if command:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv["RevCtrl"] == expected_result
+                else:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv["RevCtrl"] == False
+
+
+def test_fwd_rev_interlock_en_false():
+    for op_mode, src_mode, set_command, expected_result in test_scenario:
+        for command in [True, False]:
+            bin_drv = init_bin_drv(
+                init_op_mode=op_mode,
+                init_src_mode=src_mode,
+                fwd_en=True,
+                rev_en=True,
+                intl_en=False,
+            )
+            bin_drv.set_interlock(True)
+            eval(f"bin_drv.{set_command}({command})")
+            print(
+                f"Scenario: mode {op_mode} {src_mode}, {set_command}, expected: {expected_result}"
+            )
+
+            assert bin_drv["Interlock"] == True
+            assert bin_drv["SafePosAct"] == False
+            if set_command in ["set_fwd_op", "set_fwd_aut"]:
+                if command:
+                    assert bin_drv["FwdCtrl"] == expected_result
+                    assert bin_drv["RevCtrl"] == False
+                else:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv["RevCtrl"] == False
+            elif set_command in ["set_rev_op", "set_rev_aut"]:
+                if command:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv["RevCtrl"] == expected_result
+                else:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv["RevCtrl"] == False
+
+
+def test_fwd_rev_protect_en_true():
+    for op_mode, src_mode, set_command, _ in test_scenario:
+        for command in [True, False]:
+            bin_drv = init_bin_drv(
+                init_op_mode=op_mode,
+                init_src_mode=src_mode,
+                fwd_en=True,
+                rev_en=True,
+                prot_en=True,
+            )
+            bin_drv.set_protect(False)
+            eval(f"bin_drv.{set_command}({command})")
+            print(
+                f"Scenario: mode {op_mode} {src_mode}, {set_command}, expected: False"
+            )
+
+            assert bin_drv["Protect"] == False
+            assert bin_drv["SafePosAct"] == True
+            assert bin_drv["FwdCtrl"] == False
+            assert bin_drv["RevCtrl"] == False
+
+    for op_mode, src_mode, set_command, expected_result in test_scenario:
+        for command in [True, False]:
+            bin_drv = init_bin_drv(
+                init_op_mode=op_mode,
+                init_src_mode=src_mode,
+                fwd_en=True,
+                rev_en=True,
+                prot_en=True,
+            )
+            bin_drv.set_protect(True)
+            eval(f"bin_drv.{set_command}({command})")
+            print(
+                f"Scenario: mode {op_mode} {src_mode}, {set_command}, expected: {expected_result}"
+            )
+
+            assert bin_drv["Protect"] == True
+            assert bin_drv["SafePosAct"] == False
+            if set_command in ["set_fwd_op", "set_fwd_aut"]:
+                if command:
+                    assert bin_drv["FwdCtrl"] == expected_result
+                    assert bin_drv["RevCtrl"] == False
+                else:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv["RevCtrl"] == False
+            elif set_command in ["set_rev_op", "set_rev_aut"]:
+                if command:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv["RevCtrl"] == expected_result
+                else:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv["RevCtrl"] == False
+
+
+def test_fwd_rev_protect_en_false():
+    for op_mode, src_mode, set_command, expected_result in test_scenario:
+        for command in [True, False]:
+            bin_drv = init_bin_drv(
+                init_op_mode=op_mode,
+                init_src_mode=src_mode,
+                fwd_en=True,
+                rev_en=True,
+                prot_en=False,
+            )
+            bin_drv.set_protect(True)
+            eval(f"bin_drv.{set_command}({command})")
+            print(
+                f"Scenario: mode {op_mode} {src_mode}, {set_command}, expected: {expected_result}"
+            )
+
+            assert bin_drv["Protect"] == True
+            assert bin_drv["SafePosAct"] == False
+            if set_command in ["set_fwd_op", "set_fwd_aut"]:
+                if command:
+                    assert bin_drv["FwdCtrl"] == expected_result
+                    assert bin_drv["RevCtrl"] == False
+                else:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv["RevCtrl"] == False
+            elif set_command in ["set_rev_op", "set_rev_aut"]:
+                if command:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv["RevCtrl"] == expected_result
+                else:
+                    assert bin_drv["FwdCtrl"] == False
+                    assert bin_drv["RevCtrl"] == False
+
+
+test_scenario_reset = [
+    ("off", "int", "set_reset_op", False),
+    ("off", "man", "set_reset_op", False),
+    ("op", "int", "set_reset_op", True),
+    ("op", "man", "set_reset_op", True),
+    ("aut", "int", "set_reset_op", False),
+    ("aut", "man", "set_reset_op", False),
+    ("off", "int", "set_reset_aut", False),
+    ("off", "man", "set_reset_aut", False),
+    ("op", "int", "set_reset_aut", False),
+    ("op", "man", "set_reset_aut", False),
+    ("aut", "int", "set_reset_aut", True),
+    ("aut", "man", "set_reset_aut", True),
+]
+
+
+def test_reset():
+    for op_mode, src_mode, set_command, result in test_scenario_reset:
+        for command in [True, False]:
+            bin_drv = init_bin_drv(
+                init_op_mode=op_mode,
+                init_src_mode=src_mode,
+                fwd_en=True,
+                rev_en=True,
+                prot_en=True,
+            )
+
+            bin_drv.set_protect(False)
+            assert bin_drv["Protect"] == False
+            assert bin_drv["SafePosAct"] == True
+
+            eval(f"bin_drv.{set_command}({command})")
+            print(
+                f"Scenario: mode {op_mode} {src_mode}, {set_command}, expected: {result}"
+            )
+
+            if command:
+                assert bin_drv["Protect"] == result
+                assert bin_drv["SafePosAct"] != result
+            else:
+                assert bin_drv["Protect"] == False
+                assert bin_drv["SafePosAct"] == True
+
+
+test_scenario_stop = [
+    ("op", "int", "set_stop_op", True),
+    ("op", "man", "set_stop_op", True),
+    ("aut", "int", "set_stop_op", False),
+    ("aut", "man", "set_stop_op", False),
+    ("op", "int", "set_stop_aut", False),
+    ("op", "man", "set_stop_aut", False),
+    ("aut", "int", "set_stop_aut", True),
+    ("aut", "man", "set_stop_aut", True),
+]
+
+
+def test_stop():
+    for op_mode, src_mode, set_command, changes_expected in test_scenario_stop:
+        for command in [True, False]:
+            bin_drv = init_bin_drv(
+                init_op_mode=op_mode, init_src_mode=src_mode, fwd_en=True, rev_en=True
+            )
+            if op_mode == "op":  # open valve
+                bin_drv.set_fwd_op(True)
+            elif op_mode == "aut":
+                bin_drv.set_fwd_aut(True)
+
+            print(
+                f"Scenario: mode {op_mode} {src_mode}, {set_command}, expected: {changes_expected}"
+            )
+            if changes_expected:
+                assert bin_drv["FwdCtrl"] == True
+                eval(f"bin_drv.{set_command}({command})")
+                if command:
+                    assert bin_drv["FwdCtrl"] == False
+                else:
+                    assert bin_drv["FwdCtrl"] == True
+            else:
+                assert bin_drv["FwdCtrl"] == True
+
+
+def test_fwd_fbk():
+    bin_drv = init_bin_drv(fwd_fbk_calc=False, rev_fbk_calc=False)
+    assert bin_drv.get_fwd_fbk() == False
+    bin_drv.set_fwd_fbk(True)
+    assert bin_drv.get_fwd_fbk() == True
+
+
+def test_rev_fbk():
+    bin_drv = init_bin_drv(fwd_fbk_calc=False, rev_fbk_calc=False)
+    assert bin_drv.get_rev_fbk() == False
+    bin_drv.set_rev_fbk(True)
+    assert bin_drv.get_rev_fbk() == True
+
+
+if __name__ == "__main__":
+    # This allows running the tests directly if needed
+    pytest.main([__file__, "-v"])