feat: add uv to python build images (#181)

* feat: add uv to python>=3.10 build images

* fix: use pip to install rather than script

* nit: install uv in 3.8/3.9

* test: add tests for 3.8/3.9 for uv

Author: reedham-aws

build-image-src/Dockerfile-python310

@@ -50,6 +50,9 @@ ENV LANG=en_US.UTF-8
 # Python for it to be picked up during `sam build`
 RUN pip3 install wheel==0.45.1
 
+# Install uv
+RUN pip3 install uv==0.9.22
+
 COPY ATTRIBUTION.txt /
 
 # Compatible with initial base image

build-image-src/Dockerfile-python311

@@ -50,6 +50,9 @@ ENV LANG=en_US.UTF-8
 # Python for it to be picked up during `sam build`
 RUN pip3 install wheel==0.45.1
 
+# Install uv
+RUN pip3 install uv==0.9.22
+
 COPY ATTRIBUTION.txt /
 
 # Compatible with initial base image

build-image-src/Dockerfile-python312

@@ -51,6 +51,9 @@ ENV LANG=en_US.UTF-8
 # Python for it to be picked up during `sam build`
 RUN pip3 install wheel==0.45.1 setuptools
 
+# Install uv
+RUN pip3 install uv==0.9.22
+
 COPY ATTRIBUTION.txt /
 
 # Compatible with initial base image

build-image-src/Dockerfile-python313

@@ -51,6 +51,9 @@ ENV LANG=en_US.UTF-8
 # Python for it to be picked up during `sam build`
 RUN pip3 install wheel==0.45.1 setuptools
 
+# Install uv
+RUN pip3 install uv==0.9.22
+
 COPY ATTRIBUTION.txt /
 
 # Compatible with initial base image

build-image-src/Dockerfile-python314

@@ -51,6 +51,9 @@ ENV LANG=en_US.UTF-8
 # Python for it to be picked up during `sam build`
 RUN pip3 install wheel==0.45.1 setuptools
 
+# Install uv
+RUN pip3 install uv==0.9.22
+
 COPY ATTRIBUTION.txt /
 
 # Compatible with initial base image

build-image-src/Dockerfile-python38

@@ -50,6 +50,9 @@ ENV LANG=en_US.UTF-8
 # Python for it to be picked up during `sam build`
 RUN pip3 install wheel==0.45.1
 
+# Install uv
+RUN pip3 install uv==0.9.22
+
 COPY ATTRIBUTION.txt /
 
 # Compatible with initial base image

build-image-src/Dockerfile-python39

@@ -50,6 +50,9 @@ ENV LANG=en_US.UTF-8
 # Python for it to be picked up during `sam build`
 RUN pip3 install wheel==0.45.1
 
+# Install uv
+RUN pip3 install uv==0.9.22
+
 COPY ATTRIBUTION.txt /
 
 # Compatible with initial base image

tests/test_build_images.py

@@ -507,6 +507,7 @@ def test_packages(self):
         self.assertTrue(self.check_package_output("python --version", "Python 3.8."))
         self.assertTrue(self.is_package_present("pip"))
         self.assertTrue(self.is_architecture("x86_64"))
+        self.assertTrue(self.check_package_output("uv --version", "uv"))
 
 
 @pytest.mark.python38arm64
@@ -524,6 +525,7 @@ def test_packages(self):
         self.assertTrue(self.check_package_output("python --version", "Python 3.8."))
         self.assertTrue(self.is_package_present("pip"))
         self.assertTrue(self.is_architecture("aarch64"))
+        self.assertTrue(self.check_package_output("uv --version", "uv"))
 
 
 @pytest.mark.python39x86_64
@@ -540,6 +542,24 @@ def test_packages(self):
         """
         self.assertTrue(self.check_package_output("python --version", "Python 3.9."))
         self.assertTrue(self.is_package_present("pip"))
+        self.assertTrue(self.check_package_output("uv --version", "uv"))
+
+
+@pytest.mark.python39arm64
+class TestBIPython39ForArm(BuildImageBase):
+    __test__ = True
+
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass("python3.9", "Dockerfile-python39", "pip", tag="arm64")
+
+    def test_packages(self):
+        """
+        Test packages specific to this build image
+        """
+        self.assertTrue(self.check_package_output("python --version", "Python 3.9."))
+        self.assertTrue(self.is_package_present("pip"))
+        self.assertTrue(self.check_package_output("uv --version", "uv"))
 
 
 @pytest.mark.python310arm64
@@ -556,6 +576,7 @@ def test_packages(self):
         """
         self.assertTrue(self.check_package_output("python --version", "Python 3.10."))
         self.assertTrue(self.is_package_present("pip"))
+        self.assertTrue(self.check_package_output("uv --version", "uv"))
 
 
 @pytest.mark.python310x86_64
@@ -572,6 +593,7 @@ def test_packages(self):
         """
         self.assertTrue(self.check_package_output("python --version", "Python 3.10."))
         self.assertTrue(self.is_package_present("pip"))
+        self.assertTrue(self.check_package_output("uv --version", "uv"))
 
 
 @pytest.mark.python311arm64
@@ -588,6 +610,7 @@ def test_packages(self):
         """
         self.assertTrue(self.check_package_output("python --version", "Python 3.11."))
         self.assertTrue(self.is_package_present("pip"))
+        self.assertTrue(self.check_package_output("uv --version", "uv"))
 
 
 @pytest.mark.python311x86_64
@@ -604,6 +627,7 @@ def test_packages(self):
         """
         self.assertTrue(self.check_package_output("python --version", "Python 3.11."))
         self.assertTrue(self.is_package_present("pip"))
+        self.assertTrue(self.check_package_output("uv --version", "uv"))
 
 
 @pytest.mark.python312arm64
@@ -620,6 +644,7 @@ def test_packages(self):
         """
         self.assertTrue(self.check_package_output("python --version", "Python 3.12."))
         self.assertTrue(self.is_package_present("pip"))
+        self.assertTrue(self.check_package_output("uv --version", "uv"))
 
 
 @pytest.mark.python312x86_64
@@ -636,6 +661,7 @@ def test_packages(self):
         """
         self.assertTrue(self.check_package_output("python --version", "Python 3.12."))
         self.assertTrue(self.is_package_present("pip"))
+        self.assertTrue(self.check_package_output("uv --version", "uv"))
 
 
 @pytest.mark.python313arm64
@@ -652,6 +678,7 @@ def test_packages(self):
         """
         self.assertTrue(self.check_package_output("python --version", "Python 3.13."))
         self.assertTrue(self.is_package_present("pip"))
+        self.assertTrue(self.check_package_output("uv --version", "uv"))
 
 
 @pytest.mark.python313x86_64
@@ -668,6 +695,7 @@ def test_packages(self):
         """
         self.assertTrue(self.check_package_output("python --version", "Python 3.13."))
         self.assertTrue(self.is_package_present("pip"))
+        self.assertTrue(self.check_package_output("uv --version", "uv"))
 
 
 @pytest.mark.python314arm64
@@ -684,6 +712,7 @@ def test_packages(self):
         """
         self.assertTrue(self.check_package_output("python --version", "Python 3.14."))
         self.assertTrue(self.is_package_present("pip"))
+        self.assertTrue(self.check_package_output("uv --version", "uv"))
 
 
 @pytest.mark.python314x86_64
@@ -700,22 +729,7 @@ def test_packages(self):
         """
         self.assertTrue(self.check_package_output("python --version", "Python 3.14."))
         self.assertTrue(self.is_package_present("pip"))
-
-
-@pytest.mark.python39arm64
-class TestBIPython39ForArm(BuildImageBase):
-    __test__ = True
-
-    @classmethod
-    def setUpClass(cls):
-        super().setUpClass("python3.9", "Dockerfile-python39", "pip", tag="arm64")
-
-    def test_packages(self):
-        """
-        Test packages specific to this build image
-        """
-        self.assertTrue(self.check_package_output("python --version", "Python 3.9."))
-        self.assertTrue(self.is_package_present("pip"))
+        self.assertTrue(self.check_package_output("uv --version", "uv"))
 
 
 @pytest.mark.dotnet6x86_64