add black and mypy, also fix silly name in test

mlcommons · Dec 8, 2023 · 0acfa07 · 0acfa07
1 parent 138d69f
commit 0acfa07
Show file tree

Hide file tree

Showing 3 changed files with 131 additions and 3 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,6 +18,8 @@ jq = "^1.6.0"
 [tool.poetry.group.dev.dependencies]
 pytest-datafiles = "^3.0.0"
 pytest = "^7.4.3"
+mypy = "^1.7.1"
+black = "^23.11.0"
 
 [tool.pytest.ini_options]
 addopts = [

diff --git a/tests/test_static_site_generator.py b/tests/test_static_site_generator.py
@@ -42,5 +42,5 @@ def test_creates_files(benchmark, cwd_tmpdir):
 def test_displays_correct_stars(benchmark, cwd_tmpdir, monkeypatch, score, expected):
     monkeypatch.setattr(benchmark, "overall_score", lambda: score)
     generator = StaticSiteGenerator()
-    foo = generator.calculate_stars(benchmark)
-    assert foo == expected
+    stars = generator.calculate_stars(benchmark)
+    assert stars == expected