5b443a2da38e17d58e2c3ef4dbb364eff7affa3b

 1diff --git a/tests/test_get_venv_type.zunit b/tests/test_get_venv_type.zunit
 2new file mode 100644
 3index 0000000000000000000000000000000000000000..f1817798740486c763095db36ba48da49ff9864a
 4--- /dev/null
 5+++ b/tests/test_get_venv_type.zunit
 6@@ -0,0 +1,52 @@
 7+#!/usr/bin/env zunit
 8+
 9+@setup {
10+    load "../autoswitch_virtualenv.plugin.zsh"
11+    TARGET="$(mktemp -d)"
12+}
13+
14+@teardown {
15+    rm -rf "$TARGET"
16+}
17+
18+@test '_get_venv_type pipenv' {
19+    touch "$TARGET/Pipfile"
20+
21+    run _get_venv_type "$TARGET"
22+
23+    assert $state equals 0
24+    assert "$output" same_as "pipenv"
25+}
26+
27+
28+@test '_get_venv_type virtualenv (requirements.txt)' {
29+    touch "$TARGET/requirements.txt"
30+
31+    run _get_venv_type "$TARGET" "unknown"
32+
33+    assert $state equals 0
34+    assert "$output" same_as "virtualenv"
35+}
36+
37+@test '_get_venv_type virtualenv (setup.py)' {
38+    touch "$TARGET/setup.py"
39+
40+    run _get_venv_type "$TARGET" "unknown"
41+
42+    assert $state equals 0
43+    assert "$output" same_as "virtualenv"
44+}
45+
46+@test '_get_venv_type virtualenv (default)' {
47+    run _get_venv_type "$TARGET"
48+
49+    assert $state equals 0
50+    assert "$output" same_as "virtualenv"
51+}
52+
53+@test '_get_venv_type virtualenv (default specified)' {
54+    run _get_venv_type "$TARGET" "foobar"
55+
56+    assert $state equals 0
57+    assert "$output" same_as "foobar"
58+}