import unittest
from typing import List

from hstest.stage_test import StageTest
from hstest.test_case import SimpleTestCase, TestCase


class TesSimpleTestCase(StageTest):

    def generate(self) -> List[TestCase]:
        return [
            SimpleTestCase(stdin="123", stdout="123\n123", feedback=''),
            SimpleTestCase(stdin="567", stdout="567\n567", feedback='')
        ]


class Test(unittest.TestCase):
    def test(self):
        status, feedback = TesSimpleTestCase('main').run_tests()
        self.assertEqual('test OK', feedback)
        self.assertEqual(status, 0)

import unittest
from typing import List

from hstest.stage_test import StageTest
from hstest.test_case import SimpleTestCase, TestCase


class TesSimpleTestCase(StageTest):

    def generate(self) -> List[TestCase]:
        return [
            SimpleTestCase(stdin="123", stdout="123\n123", feedback=''),
            SimpleTestCase(stdin="567", stdout="567\n567", feedback='')
        ]


class Test(unittest.TestCase):
    def test(self):
        status, feedback = TesSimpleTestCase('main').run_tests()
        self.assertEqual('test OK', feedback)
        self.assertEqual(status, 0)
