trs/tests/test_calc.py

import unittest


from src.calc import Parser


class TestParser(Parser):

    def __init__(self, input_buffer, **kwargs):
        Parser.__init__(self, **kwargs)

        self.input_buffer = []
        self.input_position = 0

        map(self.append, input_buffer)

    def append(self, input):
        self.input_buffer.append(input + '\n')

    def read(self, nbytes):
        buffer = ''

        try:
            buffer = self.input_buffer[self.input_position]
        except IndexError:
            return ''

        self.input_position += 1

        return buffer


class TestCalc(unittest.TestCase):

    def setUp(self):
        pass

    def tearDown(self):
        pass

    def test_constructor(self):
        assert TestParser(['1+4'], keepfiles=1).run() == 5.0