basic_tests.py

import unittest
from basic import basic
import io
import sys

import contextlib

contextlib.redirect_stdout
class auto_inout:
    def __init__(self, text: str = ''):
        self.in_buffer = io.StringIO(text)
        self.out_buffer = io.StringIO()

    def __enter__(self):
        self.old_stdin = sys.stdin
        sys.stdin = self.in_buffer
        self.old_stdout = sys.stdout
        sys.stdout = self.out_buffer
        return self.out_buffer

    def __exit__(self, *_):
        sys.stdin = self.old_stdin
        sys.stdout = self.old_stdout



class PrintTests(unittest.TestCase):

    def test_print_bare(self):

        @basic
        def print_bare():
            _10. PRINT

        with auto_inout() as f:
            print_bare()

        self.assertEqual('\n', f.getvalue())

    def test_print_string(self):

        @basic
        def print_string():
            _10.PRINT("foo")

        with auto_inout() as f:
            print_string()

        self.assertEqual('foo\n', f.getvalue())

    def test_print_no_newline(self):

        @basic
        def print_string():
            _10.PRINT("foo"._)

        with auto_inout() as f:
            print_string()

        self.assertEqual('foo ', f.getvalue())


    def test_print_num(self):

        @basic
        def print_num():
            _20. PRINT(5)

        with auto_inout() as f:
            print_num()

        self.assertEqual('5\n', f.getvalue())

    def test_print_var(self):

        @basic
        def print_var():
            _10. A = 5
            _20. PRINT(A)

        with auto_inout() as f:
            print_var()

        self.assertEqual('5\n', f.getvalue())

class InputTests(unittest.TestCase):

    def test_input_bare(self):

        @basic
        def input_bare():
            _5. INPUT.A
            _10. PRINT(A)

        with auto_inout('in1') as f:
            input_bare()

        self.assertEqual('\nin1\n', f.getvalue())

    def test_input_prompt(self):

        @basic
        def input_bare():
            _5. INPUT('One').A
            _10. PRINT(A)

        with auto_inout('Two') as f:
            input_bare()

        self.assertEqual('One \nTwo\n', f.getvalue())


class IfTests(unittest.TestCase):

    def test_if(self):

        @basic
        def if_stmt():
            _10. A = 4
            _20. IF(A == 4).THEN._40
            _30. A = 3
            _40. PRINT(A._)

        with auto_inout() as f:
            if_stmt()

        self.assertEqual('4 ', f.getvalue())

    def test_if_back(self):

        @basic
        def if_back():
            _10. A = 1
            _20. A = A + 1
            _30. IF(A < 5).THEN.GOTO._20
            _40. PRINT(A)

        with auto_inout() as f:
            if_back()

        self.assertEqual('5\n', f.getvalue())


class ForTests(unittest.TestCase):

    def test_simple_for(self):

        @basic
        def simple_for():
            _10. FOR.I=1, TO,3
            _20. PRINT(I._)
            _30. NEXT.I

        with auto_inout() as f:
            simple_for()

        self.assertEqual('1 2 3 ', f.getvalue())

    def test_nested_for(self):
        @basic
        def nested_for():
            _10.FOR.I = 1, TO, 2
            _20.FOR.J = 4, TO, 5
            _22. PRINT(I,J)
            _25. NEXT.J
            _30.NEXT.I

        with auto_inout() as f:
            nested_for()

        self.assertEqual('1 4\n1 5\n2 4\n2 5\n', f.getvalue())

    # TODO: Add tests for error conditions


if __name__ == '__main__':
    unittest.main()