diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index abfc233..b3dc304 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -10,7 +10,7 @@ jobs: strategy: fail-fast: false matrix: - group: [0, 1, 2, 3, 4] + group: [0, 1, 2, 3, 4, 5_company_logos] steps: - uses: actions/checkout@v2 - name: Set up Python 3.8 diff --git a/test/integration/test_generator.py b/test/integration/test_generator.py index e03f3ce..83b776d 100644 --- a/test/integration/test_generator.py +++ b/test/integration/test_generator.py @@ -139,6 +139,21 @@ def test_generation_level_4(microservice_dir, mock_input_sequence): ) assert generator.generate() == 0 +@pytest.mark.parametrize('mock_input_sequence', [['y']], indirect=True) +def test_generation_level_5_company_logos(microservice_dir, mock_input_sequence): + os.environ['VERBOSE'] = 'true' + generator = Generator( + f'''\ +Given a list of email addresses, get all company names from them. +For all companies, get the company logo. +All logos need to be arranged on a square. +The square is returned as png. +''', + str(microservice_dir), + 'gpt-3.5-turbo' + ) + assert generator.generate() == 0 + @pytest.mark.parametrize('mock_input_sequence', [['y', 'https://upload.wikimedia.org/wikipedia/commons/thumb/4/47/PNG_transparency_demonstration_1.png/560px-PNG_transparency_demonstration_1.png']], indirect=True) def test_generation_level_5(microservice_dir, mock_input_sequence): """