test_caf_machine.py

 
import unittest
from unittest import TestCase
 
from caf.state_machines import ConditionError, Machine, MachineError
 
 
class Test_Machine(TestCase):
    """
    Set of tests for checking Machine Base class
    """
 
    def setUp(self):
        """
        """
        
        self.mm = Machine()
        self.mm.add_state("standing")
        self.mm.add_state("walking")
        self.mm.add_transition("walk", "standing", "walking")
        self.mm.add_transition("stop", "walking", "standing")
        self.mm.initial_state = "standing"
 
    def test_initial(self):
        """
        Checks that a new machine sets an initial state correctly
        """
        states = ["walking", "standing"]
        initial = "walking"
        m = Machine(states, initial)
        self.assertEqual(m.state.name, initial)
 
    def test_set(self):
        """
        Checks that machine sets at a state
        """
        self.mm.state = "walking"
        self.assertEqual(self.mm.state.name, "walking")
 
    def test_bad_set(self):
        """
        Checks that machine won't set a state not in the list
        """
        with self.assertRaises(MachineError):
            self.mm.state = "fake"
 
    def test_walk(self):
        """
        Checks that machine can transition to a state
        """
        self.mm.state = "standing"
        self.mm.walk()
        self.assertEqual(self.mm.state.name, "walking")
 
    def test_walk_stop(self):
        """
        Checks that machine can traverse several states
        """
        self.mm.state = "standing"
        self.mm.walk()
        self.mm.stop()
        self.assertEqual(self.mm.state.name, "standing")
 
    def test_true_condition(self):
        """
        Check that when a condition for a transition evaulates true it will transition
        to the correct state
        """
        self.mm.add_state("airborne")
 
        def can_jump():
            return True
 
        self.mm.add_transition("jump", "standing", "airborne", conditions=can_jump)
        self.mm.state = "standing"
        self.mm.jump()
        self.assertEqual(self.mm.state.name, "airborne")
 
    def test_false_condition(self):
        """
        Check that when a condition for a transition evaulates false it will raise
        the correct error.
        """
        self.mm.add_state("airborne")
 
        def can_jump():
            return False
 
        self.mm.add_transition("jump", "standing", "airborne", conditions=can_jump)
        self.mm.state = "standing"
        with self.assertRaises(ConditionError):
            self.mm.jump()
 
    def test_true_conditions(self):
        """
        Check that when multiple conditions for a transition all evaulate true it will transition
        to the correct state
        """
        self.mm.add_state("airborne")
 
        def can_jump():
            return True
 
        def eaten_lunch():
            return True
        self.mm.add_transition("jump", "standing", "airborne", conditions=[can_jump, eaten_lunch])
        self.mm.state = "standing"
        self.mm.jump()
        self.assertEqual(self.mm.state.name, "airborne")
 
    def test_halftrue_conditions(self):
        """
        Check that when a single condition out of many for a transition evaulates true it will
        fail to move to the correct state
        """
        self.mm.add_state("airborne")
 
        def can_jump():
            return True
 
        def eaten_lunch():
            return False
        self.mm.add_transition("jump", "standing", "airborne", conditions=[can_jump, eaten_lunch])
        self.mm.state = "standing"
        with self.assertRaises(ConditionError):
            self.mm.jump()
 
 
def main():
    unittest.main()
 
 
if __name__ == '__main__':
    main()