refactor: PR feedback

Author: yt-ms

src/structurizr/view/view_set.py

@@ -371,10 +371,7 @@ def get_view(self, key: str) -> Optional[AbstractView]:
 
     def __getitem__(self, key: str) -> AbstractView:
         """Return the view with the given key or raise a KeyError."""
-        result = self._views.get(key)
-        if not result:
-            raise KeyError(f"No view with key '{key}' in ViewSet")
-        return result
+        return self._views[key]
 
     def copy_layout_information_from(self, source: "ViewSet") -> None:
         """Copy all the layout information from a source ViewSet."""

tests/unit/view/test_view_set.py

@@ -125,7 +125,7 @@ def test_getting_view_by_key(empty_viewset):
     assert viewset.get_view("container1") is container_view
     assert viewset.get_view("bogus") is None
     assert viewset["container1"] is container_view
-    with pytest.raises(KeyError, match="No view with key"):
+    with pytest.raises(KeyError):
         viewset["bogus"]