Update diagram

Author: johnsutor

CHANGELOG.md

@@ -1,4 +1,8 @@
 # Changelog 
+## 1.3.6 
+- Add option to incorporate NeetCode shorts
+- Simplify codebase and expand testing 
+
 ## 1.3.5
 - Add Jinja templates in favor over standard library for string formatting 
 - Add problems pulled from Neetcode solutions GitHub

leetcode_study_tool/formatters.py

@@ -81,6 +81,7 @@ def render_template(
         )
         template = custom_env.get_template(custom_file)
     else:
+        assert template_name is not None
         template = env.get_template(template_name)
 
     kwargs["solution_url"] = format_solution_link

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "leetcode-study-tool"
-version = "1.3.5"
+version = "1.3.6"
 description = "A tool for studying Leetcode with Python"
 requires-python = ">=3.8"
 authors = [{name="John Sutor", email="johnsutor3@gmail.com" }]

static/leetcode_study_tool_diagram.png

@@ -1,4 +1,5 @@
 import os
+import tempfile
 import unittest
 from datetime import date
 from unittest.mock import patch
@@ -34,7 +35,7 @@ def test_save_excel(self, mock_workbook):
         mock_workbook.assert_called_once_with("fake-file.xlsx")
 
     def test_save_excel_creates_file(self):
-        test_file = "/tmp/test_leetcode_output"
+        test_file = os.path.join(tempfile.gettempdir(), "test_leetcode_output")
         try:
             outputs.save_excel([self.fake_excel_row], test_file)
             self.assertTrue(os.path.exists(test_file + ".xlsx"))
@@ -50,7 +51,7 @@ def test_save_excel_column_count_matches(self):
         )
 
     def test_save_excel_skips_none_problems(self):
-        test_file = "/tmp/test_leetcode_none"
+        test_file = os.path.join(tempfile.gettempdir(), "test_leetcode_none")
         try:
             outputs.save_excel([None, self.fake_excel_row, None], test_file)
             self.assertTrue(os.path.exists(test_file + ".xlsx"))