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', '+'])