Fix unittest for Jinja v3 (#45)

aws-cloudformation · Jan 26, 2024 · a1f00a6 · a1f00a6
1 parent 6c24b61
commit a1f00a6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/unit/utils/test_miscellaneous.py b/tests/unit/utils/test_miscellaneous.py
@@ -14,10 +14,10 @@
 def test_jinja_loader_none_spec(mock_find_spec):
     """Test jinja loader"""
     mock_find_spec.return_value = None
-    env = jinja_loader("src.rpdk.guard_rail.core")
+    env = jinja_loader("rpdk.guard_rail.core")
     assert (
         env.loader.loaders[0].package_path
-        == PackageLoader("src.rpdk.guard_rail.core").package_path
+        == PackageLoader("rpdk.guard_rail.core").package_path
     )
 
 
@@ -28,7 +28,7 @@ def test_jinja_loader(mock_file_systeam_loader):
     mock_file_systeam_loader.return_value = FileSystemLoader(
         str(spec_path.parent / "templates")
     )
-    env = jinja_loader("src.rpdk.guard_rail.core")
+    env = jinja_loader("rpdk.guard_rail.core")
     assert (
         env.loader.loaders[0].searchpath
         == FileSystemLoader(str(spec_path.parent / "templates")).searchpath