enum tests

jonathan-buttner · Mar 26, 2024 · ad5e084 · ad5e084
1 parent 463833c
commit ad5e084
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/...rg/elasticsearch/xpack/inference/services/cohere/embeddings/CohereEmbeddingTypeTests.java b/...rg/elasticsearch/xpack/inference/services/cohere/embeddings/CohereEmbeddingTypeTests.java
@@ -9,6 +9,7 @@
 
 import org.elasticsearch.TransportVersion;
 import org.elasticsearch.TransportVersions;
+import org.elasticsearch.index.mapper.vectors.DenseVectorFieldMapper;
 import org.elasticsearch.test.ESTestCase;
 
 import static org.hamcrest.Matchers.is;
@@ -48,4 +49,12 @@ public void testTranslateToVersion_ReturnsFloat_WhenVersionOnByteEnumAddition_Wh
             is(CohereEmbeddingType.FLOAT)
         );
     }
+
+    public void testFromElementType_CovertsFloatToCohereEmbeddingTypeFloat() {
+        assertThat(CohereEmbeddingType.fromElementType(DenseVectorFieldMapper.ElementType.FLOAT), is(CohereEmbeddingType.FLOAT));
+    }
+
+    public void testFromElementType_CovertsByteToCohereEmbeddingTypeByte() {
+        assertThat(CohereEmbeddingType.fromElementType(DenseVectorFieldMapper.ElementType.BYTE), is(CohereEmbeddingType.BYTE));
+    }
 }