steve/databricks-cli
steve
/
databricks-cli
mirror of https://github.com/databricks/cli.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

refactor: update tests to use t.TempDir(), uv, and pyproject.toml

This commit is contained in:
Denis Bilenko (aider) 2025-03-03 16:44:14 +01:00
parent c0cd244e22
commit 9c58b2eeaa
1 changed files with 23 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
libs/patchwheel/patchwheel_test.go
View File

@ -13,15 +13,19 @@ import (
// minimalPythonProject returns a map of file paths to their contents for a minimal Python project.
func minimalPythonProject() map[string]string {
// AI TODO: switch to pyproject.toml here
return map[string]string{
// A simple setup.py that uses setuptools.
"setup.py": `from setuptools import setup, find_packages
setup(
name="myproj",
version="0.1.0",
packages=find_packages(),
)`,
// pyproject.toml using modern Python packaging
"pyproject.toml": `[project]
name = "myproj"
version = "0.1.0"
[build-system]
requires = ["setuptools>=61.0.0", "wheel"]
build-backend = "setuptools.build_meta"
[tool.setuptools.packages.find]
where = ["."]
`,
// A simple module with a __version__.
"myproj/__init__.py": `__version__ = "0.1.0"
def hello():
@ -59,13 +63,8 @@ func TestPatchWheel(t *testing.T) {
pythonVersions := []string{"python3.9", "python3.10", "python3.11", "python3.12"}
for _, py := range pythonVersions {
t.Run(py, func(t *testing.T) {
// Create a temporary directory for the project.
// AI TODO: use t.TempDir() instead
tempDir, err := ioutil.TempDir("", "myproj")
if err != nil {
t.Fatal(err)
}
defer os.RemoveAll(tempDir)
// Create a temporary directory for the project
tempDir := t.TempDir()
// Write minimal Python project files.
projFiles := minimalPythonProject()
@ -73,11 +72,9 @@ func TestPatchWheel(t *testing.T) {
t.Fatal(err)
}
// Create a virtual environment.
venvDir := filepath.Join(tempDir, "venv")
// AI TODO: use "uv init --python <python>" to create venv
if out, err := runCmd(tempDir, py, "-m", "venv", "venv"); err != nil {
t.Fatalf("venv creation failed: %v, output: %s", err, out)
// Create a virtual environment using uv
if out, err := runCmd(tempDir, "uv", "venv", "--python", py, "venv"); err != nil {
t.Fatalf("uv venv creation failed: %v, output: %s", err, out)
}
// Determine the pip and python paths inside the venv.
@ -85,9 +82,9 @@ func TestPatchWheel(t *testing.T) {
pyExec := filepath.Join(venvBin, "python")
pipExec := filepath.Join(venvBin, "pip")
// Upgrade pip and install build.
if out, err := runCmd(tempDir, pipExec, "install", "--upgrade", "pip", "build"); err != nil {
t.Fatalf("pip install failed: %v, output: %s", err, out)
// Install build using uv
if out, err := runCmd(tempDir, pipExec, "install", "build"); err != nil {
t.Fatalf("uv pip install failed: %v, output: %s", err, out)
}
// Build the wheel.
@ -113,10 +110,9 @@ func TestPatchWheel(t *testing.T) {
}
t.Logf("origWheel=%s patchedWheel=%s", origWheel, patchedWheel)
// Install the patched wheel.
// AI TODO: use uv
if out, err := runCmd(tempDir, pipExec, "install", "--force-reinstall", patchedWheel); err != nil {
t.Fatalf("failed to install patched wheel: %v, output: %s", err, out)
// Install the patched wheel using uv
if out, err := runCmd(tempDir, pipExec, "install", "--reinstall", patchedWheel); err != nil {
t.Fatalf("failed to install patched wheel with uv: %v, output: %s", err, out)
}
// Run a small command to import the package and print its version.