honeymoose
/
OpenSearch
mirror of https://github.com/honeymoose/OpenSearch.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[ML][TEST] Readjust AutodetectMemoryLimitIT assertions 

Model memory has decreased slightly due to
elastic/machine-learning-cpp#553. This commit adjusts
the test assertions accordingly.

Original commit: elastic/x-pack-elasticsearch@ea8d403f1a
This commit is contained in:
Dimitrios Athanasiou 2018-02-12 14:51:42 +00:00
parent d5281e89b7
commit 5d1a5f27b2
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
qa/ml-native-tests/src/test/java/org/elasticsearch/xpack/ml/integration/AutodetectMemoryLimitIT.java
View File

@ -84,10 +84,10 @@ public class AutodetectMemoryLimitIT extends MlNativeAutodetectIntegTestCase {
// it is important to check that while we rejected partitions, we still managed // it is important to check that while we rejected partitions, we still managed
// to create some by fields; it shows we utilize memory in a meaningful way // to create some by fields; it shows we utilize memory in a meaningful way
// rather than creating empty partitions // rather than creating empty partitions
assertThat(modelSizeStats.getTotalPartitionFieldCount(), lessThan(1000L)); assertThat(modelSizeStats.getTotalPartitionFieldCount(), lessThan(1100L));
assertThat(modelSizeStats.getTotalPartitionFieldCount(), greaterThan(600L)); assertThat(modelSizeStats.getTotalPartitionFieldCount(), greaterThan(850L));
assertThat(modelSizeStats.getTotalByFieldCount(), lessThan(900L)); assertThat(modelSizeStats.getTotalByFieldCount(), lessThan(1000L));
assertThat(modelSizeStats.getTotalByFieldCount(), greaterThan(600L)); assertThat(modelSizeStats.getTotalByFieldCount(), greaterThan(750L));
assertThat(modelSizeStats.getMemoryStatus(), equalTo(ModelSizeStats.MemoryStatus.HARD_LIMIT)); assertThat(modelSizeStats.getMemoryStatus(), equalTo(ModelSizeStats.MemoryStatus.HARD_LIMIT));
} }
@ -133,8 +133,8 @@ public class AutodetectMemoryLimitIT extends MlNativeAutodetectIntegTestCase {
ModelSizeStats modelSizeStats = jobStats.getModelSizeStats(); ModelSizeStats modelSizeStats = jobStats.getModelSizeStats();
assertThat(modelSizeStats.getModelBytes(), lessThan(36000000L)); assertThat(modelSizeStats.getModelBytes(), lessThan(36000000L));
assertThat(modelSizeStats.getModelBytes(), greaterThan(30000000L)); assertThat(modelSizeStats.getModelBytes(), greaterThan(30000000L));
assertThat(modelSizeStats.getTotalByFieldCount(), lessThan(2100L)); assertThat(modelSizeStats.getTotalByFieldCount(), lessThan(2300L));
assertThat(modelSizeStats.getTotalByFieldCount(), greaterThan(1500L)); assertThat(modelSizeStats.getTotalByFieldCount(), greaterThan(2000L));
assertThat(modelSizeStats.getMemoryStatus(), equalTo(ModelSizeStats.MemoryStatus.HARD_LIMIT)); assertThat(modelSizeStats.getMemoryStatus(), equalTo(ModelSizeStats.MemoryStatus.HARD_LIMIT));
} }
@ -235,7 +235,7 @@ public class AutodetectMemoryLimitIT extends MlNativeAutodetectIntegTestCase {
// Assert we haven't violated the limit too much // Assert we haven't violated the limit too much
GetJobsStatsAction.Response.JobStats jobStats = getJobStats(job.getId()).get(0); GetJobsStatsAction.Response.JobStats jobStats = getJobStats(job.getId()).get(0);
ModelSizeStats modelSizeStats = jobStats.getModelSizeStats(); ModelSizeStats modelSizeStats = jobStats.getModelSizeStats();
assertThat(modelSizeStats.getModelBytes(), lessThan(95000000L)); assertThat(modelSizeStats.getModelBytes(), lessThan(90000000L));
assertThat(modelSizeStats.getModelBytes(), greaterThan(75000000L)); assertThat(modelSizeStats.getModelBytes(), greaterThan(75000000L));
assertThat(modelSizeStats.getTotalOverFieldCount(), greaterThan(140000L)); assertThat(modelSizeStats.getTotalOverFieldCount(), greaterThan(140000L));
assertThat(modelSizeStats.getMemoryStatus(), equalTo(ModelSizeStats.MemoryStatus.OK)); assertThat(modelSizeStats.getMemoryStatus(), equalTo(ModelSizeStats.MemoryStatus.OK));