@@ -151,8 +151,6 @@ public async Task MLPerfExecutorInitializesWorkloadAsExpected()
151
151
List < string > expectedCommands = new List < string >
152
152
{
153
153
"sudo usermod -aG docker anyuser" ,
154
- "sudo bash -c \" export MLPERF_SCRATCH_PATH=/dev/sdd1/scratch" +
155
- " && mkdir $MLPERF_SCRATCH_PATH/data $MLPERF_SCRATCH_PATH/models $MLPERF_SCRATCH_PATH/preprocessed_data\" " ,
156
154
"sudo -u anyuser bash -c \" make prebuild MLPERF_SCRATCH_PATH=/dev/sdd1/scratch\" " ,
157
155
"sudo docker ps" ,
158
156
"sudo docker exec -u anyuser mlperf-inference-anyuser-x86_64 sudo bash -c " +
@@ -196,6 +194,18 @@ public async Task MLPerfExecutorInitializesWorkloadAsExpected()
196
194
} )
197
195
. ReturnsAsync ( makeFileString ) ;
198
196
197
+ IEnumerable < string > expectedDirectories = new List < string >
198
+ {
199
+ "/dev/sdd1/scratch/data" ,
200
+ "/dev/sdd1/scratch/models" ,
201
+ "/dev/sdd1/scratch/preprocessed_data"
202
+ } ;
203
+
204
+ this . mockFixture . Directory . Setup ( d => d . CreateDirectory ( It . IsAny < string > ( ) ) ) . Callback < string > ( ( directory ) =>
205
+ {
206
+ Assert . IsTrue ( expectedDirectories . Select ( ed => directory . Contains ( ed ) ) . Any ( ) ) ;
207
+ } ) ;
208
+
199
209
List < string > commandsExecuted = new List < string > ( ) ;
200
210
this . mockFixture . ProcessManager . OnCreateProcess = ( file , arguments , workingDirectory ) =>
201
211
{
@@ -246,6 +256,17 @@ public async Task MLPerfExecutorExecutesAsExpected()
246
256
. ReturnsAsync ( makeFileString ) ;
247
257
248
258
IEnumerable < string > expectedCommands = this . GetExpectedCommands ( ) ;
259
+ IEnumerable < string > expectedDirectories = new List < string >
260
+ {
261
+ "/dev/sdd1/scratch/data" ,
262
+ "/dev/sdd1/scratch/models" ,
263
+ "/dev/sdd1/scratch/preprocessed_data"
264
+ } ;
265
+
266
+ this . mockFixture . Directory . Setup ( d => d . CreateDirectory ( It . IsAny < string > ( ) ) ) . Callback < string > ( ( directory ) =>
267
+ {
268
+ Assert . IsTrue ( expectedDirectories . Select ( ed => directory . Contains ( ed ) ) . Any ( ) ) ;
269
+ } ) ;
249
270
250
271
List < string > commandsExecuted = new List < string > ( ) ;
251
272
@@ -298,8 +319,6 @@ private IEnumerable<string> GetExpectedCommands()
298
319
commands = new List < string >
299
320
{
300
321
"sudo usermod -aG docker anyuser" ,
301
- "sudo bash -c \" export MLPERF_SCRATCH_PATH=/dev/sdd1/scratch" +
302
- " && mkdir $MLPERF_SCRATCH_PATH/data $MLPERF_SCRATCH_PATH/models $MLPERF_SCRATCH_PATH/preprocessed_data\" " ,
303
322
"sudo -u anyuser bash -c \" make prebuild MLPERF_SCRATCH_PATH=/dev/sdd1/scratch\" " ,
304
323
"sudo docker ps" ,
305
324
"sudo docker exec -u anyuser mlperf-inference-anyuser-x86_64 sudo bash -c " +
0 commit comments