diff --git a/test/pytorch/hello.sh b/test/pytorch/hello.sh
new file mode 100644
index 0000000..d5997c5
--- /dev/null
+++ b/test/pytorch/hello.sh
@@ -0,0 +1,9 @@
+#!/bin/bash
+
+set -e
+
+source dev-container-features-test-lib
+
+check "import torch" python -c 'import torch'
+
+reportResults
diff --git a/test/pytorch/scenarios.json b/test/pytorch/scenarios.json
new file mode 100644
index 0000000..918028a
--- /dev/null
+++ b/test/pytorch/scenarios.json
@@ -0,0 +1,10 @@
+{
+    "empty": {
+        "image": "mcr.microsoft.com/devcontainers/base:ubuntu",
+        "features": {
+            "pytorch": {
+                "import": "torch"
+            }
+        }
+    }
+}
diff --git a/test/pytorch/test.sh b/test/pytorch/test.sh
new file mode 100644
index 0000000..9b78959
--- /dev/null
+++ b/test/pytorch/test.sh
@@ -0,0 +1,48 @@
+#!/bin/bash
+
+# This test file will be executed against an auto-generated devcontainer.json that
+# includes the 'hello' Feature with no options.
+#
+# For more information, see: https://github.com/devcontainers/cli/blob/main/docs/features/test.md
+#
+# Eg:
+# {
+#    "image": "<..some-base-image...>",
+#    "features": {
+#      "hello": {}
+#    },
+#    "remoteUser": "root"
+# }
+#
+# Thus, the value of all options will fall back to the default value in
+# the Feature's 'devcontainer-feature.json'.
+# For the 'hello' feature, that means the default favorite greeting is 'hey'.
+#
+# These scripts are run as 'root' by default. Although that can be changed
+# with the '--remote-user' flag.
+#
+# This test can be run with the following command:
+#
+#    devcontainer features test \
+#                   --features hello   \
+#                   --remote-user root \
+#                   --skip-scenarios   \
+#                   --base-image mcr.microsoft.com/devcontainers/base:ubuntu \
+#                   /path/to/this/repo
+
+set -e
+
+# Optional: Import test library bundled with the devcontainer CLI
+# See https://github.com/devcontainers/cli/blob/HEAD/docs/features/test.md#dev-container-features-test-lib
+# Provides the 'check' and 'reportResults' commands.
+source dev-container-features-test-lib
+
+# Feature-specific tests
+# The 'check' command comes from the dev-container-features-test-lib. Syntax is...
+# check <LABEL> <cmd> [args...]
+# check "execute command" bash -c "hello | grep 'hey, $(whoami)!'"
+check "import torch" python -c 'import torch'
+
+# Report results
+# If any of the checks above exited with a non-zero exit code, the test will fail.
+reportResults
\ No newline at end of file