From f3f946b7840563bfbcda644cb1f9b15620729a73 Mon Sep 17 00:00:00 2001
From: Andrew Morgan <andrew@amorgan.xyz>
Date: Sun, 10 Jan 2021 21:54:22 -0500
Subject: [PATCH] Additional linting for unit tests

---
 tests/config/test_config.py | 33 +++++++++++++++++++++------------
 1 file changed, 21 insertions(+), 12 deletions(-)

diff --git a/tests/config/test_config.py b/tests/config/test_config.py
index 0e8eaa8..c6146af 100644
--- a/tests/config/test_config.py
+++ b/tests/config/test_config.py
@@ -1,5 +1,6 @@
-from typing import Dict
 import unittest
+from typing import Dict
+
 from my_project_name.config import Config
 from my_project_name.errors import ConfigError
 
@@ -14,12 +15,7 @@ class ConfigTestCase(unittest.TestCase):
         """Test that Config._get_cfg works correctly"""
 
         # Here's our test dictionary. Pretend that this was parsed from a YAML config file.
-        test_config_dict = {
-            "a_key": 5,
-            "some_key": {
-                "some_other_key": "some_value"
-            }
-        }
+        test_config_dict = {"a_key": 5, "some_key": {"some_other_key": "some_value"}}
 
         # We create a fake config class. This class is needed as _get_cfg pulls config options
         # from 'self.config_dict'. So, we make a class that has a self.config_dict, and fill it
@@ -43,7 +39,11 @@ class ConfigTestCase(unittest.TestCase):
 
         # Test that the value provided by the default option is used when a key does not exist
         self.assertEqual(
-            Config._get_cfg(fake_cfg, ["a_made_up_key", "this_does_not_exist"], default="The default"),
+            Config._get_cfg(
+                fake_cfg,
+                ["a_made_up_key", "this_does_not_exist"],
+                default="The default",
+            ),
             "The default",
         )
 
@@ -55,22 +55,31 @@ class ConfigTestCase(unittest.TestCase):
 
         # Test that keys that do not exist raise a ConfigError when the required argument is True
         with self.assertRaises(ConfigError):
-            Config._get_cfg(fake_cfg, ["a_made_up_key", "this_does_not_exist"], required=True)
+            Config._get_cfg(
+                fake_cfg, ["a_made_up_key", "this_does_not_exist"], required=True
+            )
 
         # Test that a ConfigError is not returned when a non-existent key is provided and required is False
         self.assertIsNone(
-            Config._get_cfg(fake_cfg, ["a_made_up_key", "this_does_not_exist"], required=False)
+            Config._get_cfg(
+                fake_cfg, ["a_made_up_key", "this_does_not_exist"], required=False
+            )
         )
 
         # Test that default is used for non-existent keys, even if required is True
         # (Typically one shouldn't use a default with required=True anyways...)
         self.assertEqual(
-            Config._get_cfg(fake_cfg, ["a_made_up_key", "this_does_not_exist"], default="something", required=True),
+            Config._get_cfg(
+                fake_cfg,
+                ["a_made_up_key", "this_does_not_exist"],
+                default="something",
+                required=True,
+            ),
             "something",
         )
 
     # TODO: Test creating a test yaml file, passing the path to Config and _parse_config_values is called correctly
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     unittest.main()