test_scaffold.py

import unittest
import os
from click.testing import CliRunner

from mesa.main import cli


class ScaffoldTest(unittest.TestCase):
    """
    Test mesa project scaffolding command
    """

    @classmethod
    def setUpClass(cls):
        cls.runner = CliRunner()

    def test_scaffold_creates_project_dir(self):
        with self.runner.isolated_filesystem():
            assert not os.path.isdir("example_project")
            self.runner.invoke(cli, ['startproject', '--no-input'])
            assert os.path.isdir("example_project")