Apply suggestions from code review

Author: MaxymVlasov

.vscode/settings.json

@@ -8,6 +8,7 @@
   },
   "markdown.validate.enabled": true,
   "python.analysis.extraPaths": [
-    "./src"
+    "./src",
+    "./tests/pytest"
   ],
 }