Correctly handle install package[extras] (#50)

* Correctly handle `install package[extras]`

* Corrected test assert to apply to both uv and pip

* Fewer tests with pip

---------

Co-authored-by: Zoran Simic <zsimic@netflix.com>

Author: zsimic

src/pickley/__init__.py

@@ -222,7 +222,6 @@ def resolve(self, settings: "TrackedSettings"):
         pip_spec = self.given_package_spec
         canonical_name, version = CFG.despecced(self.given_package_spec)
         if version:
-            pip_spec = f"{canonical_name}=={version}"
             self.resolution_reason = "pinned"
             if canonical_name == bstrap.PICKLEY:
                 self._set_canonical(canonical_name, version)
@@ -276,7 +275,9 @@ def resolve(self, settings: "TrackedSettings"):
             line = lines[0]
             canonical_name, version = CFG.despecced(line)
             if version:
-                self.pip_spec = [f"{canonical_name}=={version}"]
+                if canonical_name == self.given_package_spec:
+                    self.pip_spec = [f"{canonical_name}=={version}"]
+
                 if not self.resolution_reason:
                     self.resolution_reason = "package spec"
 

tests/test_ops.py

@@ -177,6 +177,11 @@ def test_install_pypi(cli):
     assert cli.failed
     assert "mgit was not installed with pickley" in cli.logged
 
+    if bstrap.USE_UV:
+        cli.run("-nv install yq[test]")
+        assert cli.succeeded
+        assert "Would run: uv -q pip install yq[test]" in cli.logged
+
     # Simulate a few older versions to exercise grooming
     runez.touch(".pk/mgit-/bin/mgit", logger=None)  # Simulate a buggy old installation
     runez.touch(".pk/mgit-0.9rc5+local/bin/mgit", logger=None)