tests.py

import os
import subprocess
import unittest
import unittest.mock


class Test3(unittest.TestCase):
    def test_simple(self):
        env = {k: v for k, v in os.environ.items() if not k.endswith("_COLORS")}
        env["PYTHONUTF8"] = "1"
        process = subprocess.run(
            os.path.abspath("3"),
            shell=True,
            env=env,
            stdout=subprocess.PIPE,
        )
        self.assertEqual(process.returncode, 0)
        self.assertEqual(
            process.stdout,
            """
.
├── .github
│   └── workflows
│       ├── black.yml
│       ├── macos.yml
│       ├── setup.yml
│       ├── ubuntu.yml
│       └── windows.yml
├── .gitignore
├── 3
├── 3.cmd
├── examples
│   └── 3.png
├── LICENSE.md
├── pyproject.toml
├── README.md
├── requirements.txt
├── setup.py
└── tests.py

3 directories, 15 files
""".lstrip().encode(),
        )