school/cs235/lab04/shunting/test_shunt.py

import unittest

from shunting import shunt


class ShuntTest(unittest.TestCase):

    def test_simple_add(self):
        o, s = shunt(['4', '+', '3'])
        self.assertEquals(o, ['4', '3', '+'])