diff --git a/ml-algorithms/src/test/java/org/opensearch/ml/engine/MLEngineClassLoaderTests.java b/ml-algorithms/src/test/java/org/opensearch/ml/engine/MLEngineClassLoaderTests.java index 082d808319..24fc9f5f01 100644 --- a/ml-algorithms/src/test/java/org/opensearch/ml/engine/MLEngineClassLoaderTests.java +++ b/ml-algorithms/src/test/java/org/opensearch/ml/engine/MLEngineClassLoaderTests.java @@ -16,8 +16,10 @@ import org.opensearch.ml.common.FunctionName; import org.opensearch.ml.common.input.Input; import org.opensearch.ml.common.input.execute.samplecalculator.LocalSampleCalculatorInput; +import org.opensearch.ml.common.input.parameter.MLAlgoParams; import org.opensearch.ml.common.output.Output; import org.opensearch.ml.common.output.execute.samplecalculator.LocalSampleCalculatorOutput; +import org.opensearch.ml.engine.algorithms.metrics_correlation.MetricsCorrelation; import org.opensearch.ml.engine.algorithms.sample.LocalSampleCalculator; import org.opensearch.transport.client.Client; @@ -81,4 +83,10 @@ public void testInitInstance_ClassNotFound() { MLEngineClassLoader.initInstance("SOMETHING ELSE", null, Object.class); } + @Test + public void initInstanceMetricsCorrelationReturnsNull() { + MetricsCorrelation instance = MLEngineClassLoader.initInstance(FunctionName.METRICS_CORRELATION, null, MLAlgoParams.class); + assertNull(instance); + } + }