File tree Expand file tree Collapse file tree
StabilityMatrix.Tests/Core Expand file tree Collapse file tree Original file line number Diff line number Diff 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 {
You can’t perform that action at this time.
0 commit comments