Skip to content

Commit 61ce94b

Browse files
authored
Add test for environment marker in requirements.txt
1 parent 9237ad7 commit 61ce94b

1 file changed

Lines changed: 20 additions & 0 deletions

File tree

StabilityMatrix.Tests/Core/PipInstallArgsTests.cs

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,26 @@ public void TestParsedFromRequirementsTxt()
8181
Assert.AreEqual("torch~=2.0.0 torchvision --extra-index-url https://example.org", args.ToString());
8282
}
8383

84+
[TestMethod]
85+
public void TestParsedFromRequirementsTxt_KeepsEnvironmentMarkerRequirementAsSingleArgument()
86+
{
87+
const string requirements = """
88+
onnxruntime-gpu==1.22.0; python_version < "3.11"
89+
""";
90+
91+
var args = new PipInstallArgs().WithParsedFromRequirementsTxt(requirements).ToProcessArgs();
92+
93+
Assert.AreEqual(1, args.Count());
94+
Assert.AreEqual(
95+
"\"onnxruntime-gpu==1.22.0; python_version < \\\"3.11\\\"\"",
96+
args.Single().GetQuotedValue()
97+
);
98+
Assert.AreEqual(
99+
"\"onnxruntime-gpu==1.22.0; python_version < \\\"3.11\\\"\"",
100+
args.ToString()
101+
);
102+
}
103+
84104
[TestMethod]
85105
public void TestWithUserOverrides()
86106
{

0 commit comments

Comments
 (0)