assertInnerNvidiaSMI checks that nvidia-smi runs successfully in the inner container.
(ctx context.Context, t *testing.T, containerID string)
| 197 | // assertInnerNvidiaSMI checks that nvidia-smi runs successfully in the inner |
| 198 | // container. |
| 199 | func assertInnerNvidiaSMI(ctx context.Context, t *testing.T, containerID string) { |
| 200 | t.Helper() |
| 201 | // Assert that we can run nvidia-smi in the inner container. |
| 202 | out, err := execContainerCmd(ctx, t, containerID, "docker", "exec", "workspace_cvm", "nvidia-smi") |
| 203 | require.NoError(t, err, "failed to run nvidia-smi in the inner container") |
| 204 | require.Contains(t, out, "NVIDIA-SMI", "nvidia-smi output does not contain NVIDIA-SMI") |
| 205 | } |
| 206 | |
| 207 | // startEnvboxCmd starts the envbox container with the given arguments. |
| 208 | // Ideally we would use ory/dockertest for this, but it doesn't support |
no test coverage detected