@@ -76,7 +76,7 @@ public void testExtractAcceleratorId() {
76
76
public void testExtractAccelerators () {
77
77
List <JsonObject > accelerators = appleUtil .extractAccelerators (sampleOutputJson );
78
78
79
- assertEquals (accelerators .size (), 7 );
79
+ assertEquals (accelerators .size (), 1 );
80
80
assertEquals (accelerators .get (0 ).get ("sppci_model" ).getAsString (), "Apple M1" );
81
81
}
82
82
@@ -88,7 +88,7 @@ public void testSmiOutputToUpdatedAccelerators() {
88
88
ArrayList <Accelerator > updatedAccelerators =
89
89
appleUtil .smiOutputToUpdatedAccelerators (sampleOutputJson .toString (), parsedGpuIds );
90
90
91
- assertEquals (updatedAccelerators .size (), 7 );
91
+ assertEquals (updatedAccelerators .size (), 1 );
92
92
Accelerator accelerator = updatedAccelerators .get (0 );
93
93
assertEquals (accelerator .getAcceleratorModel (), "Apple M1" );
94
94
assertEquals (accelerator .getVendor (), AcceleratorVendor .APPLE );
@@ -112,7 +112,7 @@ public String[] getUtilizationSmiCommand() {
112
112
ArrayList <Accelerator > availableAccelerators =
113
113
spyAppleUtil .getAvailableAccelerators (availableAcceleratorIds );
114
114
115
- assertEquals (availableAccelerators .size (), 7 );
115
+ assertEquals (availableAccelerators .size (), 1 );
116
116
Accelerator accelerator = availableAccelerators .get (0 );
117
117
assertEquals (accelerator .getAcceleratorModel (), "Apple M1" );
118
118
assertEquals (accelerator .getVendor (), AcceleratorVendor .APPLE );
0 commit comments