Gecko [ theprintave.ca ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2024-02-16 18:51
__init__.py	0 B	-rw-r--r--	2018-01-11 09:16
test_core.py	2.77 KB	-rw-r--r--	2019-08-11 23:02
test_discover.py	21.33 KB	-rw-r--r--	2018-01-11 09:16
test_methodical.py	18.41 KB	-rw-r--r--	2018-06-04 05:39
test_trace.py	3.2 KB	-rw-r--r--	2018-01-11 09:16
test_visualize.py	13.42 KB	-rw-r--r--	2018-01-11 09:16

Rename

from unittest import TestCase
from .._methodical import MethodicalMachine

class SampleObject(object):
    mm = MethodicalMachine()

@mm.state(initial=True)
    def begin(self):
        "initial state"
    @mm.state()
    def middle(self):
        "middle state"
    @mm.state()
    def end(self):
        "end state"

@mm.input()
    def go1(self):
        "sample input"
    @mm.input()
    def go2(self):
        "sample input"
    @mm.input()
    def back(self):
        "sample input"

@mm.output()
    def out(self):
        "sample output"

setTrace = mm._setTrace

begin.upon(go1, middle, [out])
    middle.upon(go2, end, [out])
    end.upon(back, middle, [])
    middle.upon(back, begin, [])

class TraceTests(TestCase):
    def test_only_inputs(self):
        traces = []
        def tracer(old_state, input, new_state):
            traces.append((old_state, input, new_state))
            return None # "I only care about inputs, not outputs"
        s = SampleObject()
        s.setTrace(tracer)

s.go1()
        self.assertEqual(traces, [("begin", "go1", "middle"),
                                  ])

s.go2()
        self.assertEqual(traces, [("begin", "go1", "middle"),
                                  ("middle", "go2", "end"),
                                  ])
        s.setTrace(None)
        s.back()
        self.assertEqual(traces, [("begin", "go1", "middle"),
                                  ("middle", "go2", "end"),
                                  ])
        s.go2()
        self.assertEqual(traces, [("begin", "go1", "middle"),
                                  ("middle", "go2", "end"),
                                  ])

def test_inputs_and_outputs(self):
        traces = []
        def tracer(old_state, input, new_state):
            traces.append((old_state, input, new_state, None))
            def trace_outputs(output):
                traces.append((old_state, input, new_state, output))
            return trace_outputs # "I care about outputs too"
        s = SampleObject()
        s.setTrace(tracer)

s.go1()
        self.assertEqual(traces, [("begin", "go1", "middle", None),
                                  ("begin", "go1", "middle", "out"),
                                  ])

s.go2()
        self.assertEqual(traces, [("begin", "go1", "middle", None),
                                  ("begin", "go1", "middle", "out"),
                                  ("middle", "go2", "end", None),
                                  ("middle", "go2", "end", "out"),
                                  ])
        s.setTrace(None)
        s.back()
        self.assertEqual(traces, [("begin", "go1", "middle", None),
                                  ("begin", "go1", "middle", "out"),
                                  ("middle", "go2", "end", None),
                                  ("middle", "go2", "end", "out"),
                                  ])
        s.go2()
        self.assertEqual(traces, [("begin", "go1", "middle", None),
                                  ("begin", "go1", "middle", "out"),
                                  ("middle", "go2", "end", None),
                                  ("middle", "go2", "end", "out"),
                                  ])