FM / _test

/usr/lib/python3/dist-packages/automat/_test/

/usr/lib/python3/dist-packages/automat/_test

Name	Size	Mode	Actions
__pycache__/	-	0755	rm
test_core.py	2833	0644	edit dl rm
test_discover.py	21846	0644	edit dl rm
test_methodical.py	18856	0644	edit dl rm
test_trace.py	3279	0644	edit dl rm
test_visualize.py	13744	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /usr/lib/python3/dist-packages/automat/_test/test_trace.py (3279B)

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"),
                                  ])

cmd: