Skip to content

Commit 5d8f06b

Browse files
committed
Adjust memory limits in Java REST test code
1 parent 98a7bdd commit 5d8f06b

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

x-pack/plugin/ml/qa/native-multi-node-tests/src/javaRestTest/java/org/elasticsearch/xpack/ml/integration/AutodetectMemoryLimitIT.java

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ public void testTooManyPartitions() throws Exception {
7878
// Assert we haven't violated the limit too much
7979
GetJobsStatsAction.Response.JobStats jobStats = getJobStats(job.getId()).get(0);
8080
ModelSizeStats modelSizeStats = jobStats.getModelSizeStats();
81-
assertThat(modelSizeStats.getModelBytes(), lessThan(32000000L));
81+
assertThat(modelSizeStats.getModelBytes(), lessThan(50200000L));
8282
assertThat(modelSizeStats.getModelBytes(), greaterThan(24000000L));
8383
assertThat(
8484
modelSizeStats.getMemoryStatus(),
@@ -125,7 +125,7 @@ public void testTooManyByFields() throws Exception {
125125
// Assert we haven't violated the limit too much
126126
GetJobsStatsAction.Response.JobStats jobStats = getJobStats(job.getId()).get(0);
127127
ModelSizeStats modelSizeStats = jobStats.getModelSizeStats();
128-
assertThat(modelSizeStats.getModelBytes(), lessThan(35000000L));
128+
assertThat(modelSizeStats.getModelBytes(), lessThan(45000000L));
129129
assertThat(modelSizeStats.getModelBytes(), greaterThan(25000000L));
130130
assertThat(modelSizeStats.getMemoryStatus(), equalTo(ModelSizeStats.MemoryStatus.HARD_LIMIT));
131131
}
@@ -176,7 +176,7 @@ public void testTooManyByAndOverFields() throws Exception {
176176
// Assert we haven't violated the limit too much
177177
GetJobsStatsAction.Response.JobStats jobStats = getJobStats(job.getId()).get(0);
178178
ModelSizeStats modelSizeStats = jobStats.getModelSizeStats();
179-
assertThat(modelSizeStats.getModelBytes(), lessThan(33000000L));
179+
assertThat(modelSizeStats.getModelBytes(), lessThan(71000000L));
180180
assertThat(modelSizeStats.getModelBytes(), greaterThan(24000000L));
181181
assertThat(modelSizeStats.getMemoryStatus(), equalTo(ModelSizeStats.MemoryStatus.HARD_LIMIT));
182182
}
@@ -226,7 +226,7 @@ public void testManyDistinctOverFields() throws Exception {
226226
GetJobsStatsAction.Response.JobStats jobStats = getJobStats(job.getId()).get(0);
227227
ModelSizeStats modelSizeStats = jobStats.getModelSizeStats();
228228
assertThat(modelSizeStats.getModelBytes(), lessThan(120500000L));
229-
assertThat(modelSizeStats.getModelBytes(), greaterThan(90000000L));
229+
assertThat(modelSizeStats.getModelBytes(), greaterThan(70000000L));
230230
assertThat(modelSizeStats.getMemoryStatus(), equalTo(ModelSizeStats.MemoryStatus.HARD_LIMIT));
231231
}
232232

0 commit comments

Comments
 (0)