Set up pytest

1 files changed, 29 insertions, 0 deletions

diff --git a/neovim/.config/nvim/lua/plugins/pytest.lua b/neovim/.config/nvim/lua/plugins/pytest.lua
new file mode 100644
index 0000000..20e246e
--- /dev/null
+++ b/neovim/.config/nvim/lua/plugins/pytest.lua
@@ -0,0 +1,29 @@
+return {
+  {
+    "richardhapb/pytest.nvim",
+    dependencies = { "nvim-treesitter/nvim-treesitter" },
+    opts = {}, -- Define the options here
+    config = function(_, opts)
+      require('nvim-treesitter.configs').setup {
+        ensure_installed = { 'python', 'xml' },
+        docker = {
+          enabled = function()
+            return vim.fn.getcwd():match("docker-compose.dev.yml$")
+          end,
+          enable_docker_compose = function()
+            return vim.fn.getcwd():match("docker-compose.dev.yml$")
+          end,
+          docker_compose_file = "docker-compose.dev.yml",
+        },
+        django = {
+          enabled = function()
+            return vim.fn.getcwd():match("manage.py$")
+          end,
+          django_settings_module = "krotosstudio.settings",
+        },
+      }
+
+      require('pytest').setup(opts)
+    end
+  }
+}