forked from filipg/pjfz-2020
50 lines
1.1 KiB
Python
50 lines
1.1 KiB
Python
|
#!/usr/bin/python
|
||
|
# -*- coding: utf-8 -*-
|
||
|
|
||
|
"""
|
||
|
Zadanie 202
|
||
|
|
||
|
Napisz funkcję `get_number_of_final_states(automaton)`, która oblicza
|
||
|
liczbę stanów końcowych automatu. Należy używać metod klasy (nie
|
||
|
odwoływać się bezpośrednio do pól!).
|
||
|
|
||
|
|
||
|
NAME: get_number_of_final_states
|
||
|
PARAMS: automaton
|
||
|
RETURN: int
|
||
|
POINTS: 2
|
||
|
"""
|
||
|
|
||
|
import unittest
|
||
|
from Task202 import get_number_of_final_states
|
||
|
from deterministic_automaton import DeterministicAutomaton
|
||
|
|
||
|
class Task202Test(unittest.TestCase):
|
||
|
"""Testy do zadania 202"""
|
||
|
|
||
|
def test_simple(self):
|
||
|
"""Prosty test."""
|
||
|
|
||
|
automaton = DeterministicAutomaton()
|
||
|
|
||
|
state_a = automaton.add_state()
|
||
|
state_b = automaton.add_state()
|
||
|
state_c = automaton.add_state()
|
||
|
|
||
|
automaton.mark_as_initial(state_b)
|
||
|
|
||
|
self.assertEqual(get_number_of_final_states(automaton), 0)
|
||
|
|
||
|
automaton.mark_as_final(state_a)
|
||
|
|
||
|
self.assertEqual(get_number_of_final_states(automaton), 1)
|
||
|
|
||
|
automaton.mark_as_final(state_c)
|
||
|
automaton.mark_as_final(state_b)
|
||
|
|
||
|
self.assertEqual(get_number_of_final_states(automaton), 3)
|
||
|
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
unittest.main()
|