Added unit tests

milvus-io · Apr 19, 2024 · ae8675f · ae8675f
1 parent f031efe
commit ae8675f
Show file tree

Hide file tree

Showing 2 changed files with 139 additions and 2 deletions.
diff --git a/Milvus.Client.Tests/FieldTests.cs b/Milvus.Client.Tests/FieldTests.cs
@@ -67,4 +67,141 @@ public void CreateFloatVectorTest()
         Assert.Equal(2, field.RowCount);
         Assert.Equal(2, field.Data.Count);
     }
+
+    [Fact]
+    public void CreateInt8ArrayTest()
+    {
+        var field = FieldData.CreateArray(
+            "vector",
+            new sbyte[][]
+            {
+                [1, 2],
+                [3, 4],
+            });
+
+        Assert.Equal(MilvusDataType.Array, field.DataType);
+        Assert.Equal(MilvusDataType.Int8, field.ElementType);
+        Assert.Equal(2, field.RowCount);
+        Assert.Equal(2, field.Data.Count);
+    }
+
+    [Fact]
+    public void CreateInt16ArrayTest()
+    {
+        var field = FieldData.CreateArray(
+            "vector",
+            new short[][]
+            {
+                [1, 2],
+                [3, 4],
+            });
+
+        Assert.Equal(MilvusDataType.Array, field.DataType);
+        Assert.Equal(MilvusDataType.Int16, field.ElementType);
+        Assert.Equal(2, field.RowCount);
+        Assert.Equal(2, field.Data.Count);
+    }
+
+    [Fact]
+    public void CreateInt32ArrayTest()
+    {
+        var field = FieldData.CreateArray(
+            "vector",
+            new int[][]
+            {
+                [1, 2],
+                [3, 4],
+            });
+
+        Assert.Equal(MilvusDataType.Array, field.DataType);
+        Assert.Equal(MilvusDataType.Int32, field.ElementType);
+        Assert.Equal(2, field.RowCount);
+        Assert.Equal(2, field.Data.Count);
+    }
+
+    [Fact]
+    public void CreateInt64ArrayTest()
+    {
+        var field = FieldData.CreateArray(
+            "vector",
+            new long[][]
+            {
+                [1, 2],
+                [3, 4],
+            });
+
+        Assert.Equal(MilvusDataType.Array, field.DataType);
+        Assert.Equal(MilvusDataType.Int64, field.ElementType);
+        Assert.Equal(2, field.RowCount);
+        Assert.Equal(2, field.Data.Count);
+    }
+
+    [Fact]
+    public void CreateBoolArrayTest()
+    {
+        var field = FieldData.CreateArray(
+            "vector",
+            new bool[][]
+            {
+                [true, false],
+                [false, false],
+            });
+
+        Assert.Equal(MilvusDataType.Array, field.DataType);
+        Assert.Equal(MilvusDataType.Bool, field.ElementType);
+        Assert.Equal(2, field.RowCount);
+        Assert.Equal(2, field.Data.Count);
+    }
+
+    [Fact]
+    public void CreateFloatArrayTest()
+    {
+        var field = FieldData.CreateArray(
+            "vector",
+            new float[][]
+            {
+                [1, 2],
+                [3, 4],
+            });
+
+        Assert.Equal(MilvusDataType.Array, field.DataType);
+        Assert.Equal(MilvusDataType.Float, field.ElementType);
+        Assert.Equal(2, field.RowCount);
+        Assert.Equal(2, field.Data.Count);
+    }
+
+    [Fact]
+    public void CreateDoubleArrayTest()
+    {
+        var field = FieldData.CreateArray(
+            "vector",
+            new double[][]
+            {
+                [1, 2],
+                [3, 4],
+            });
+
+        Assert.Equal(MilvusDataType.Array, field.DataType);
+        Assert.Equal(MilvusDataType.Double, field.ElementType);
+        Assert.Equal(2, field.RowCount);
+        Assert.Equal(2, field.Data.Count);
+    }
+
+    //TODO: differentiate VarChar and String somehow
+    [Fact]
+    public void CreateVarCharArrayTest()
+    {
+        var field = FieldData.CreateArray(
+            "vector",
+            new string[][]
+            {
+                ["3d4d387208e04a9abe77be65e2b7c7b3", "a5502ddb557047968a70ff69720d2dd2"],
+                ["4c246789a91f4b15aa3b26799df61457", "00a23e95823b4f14854ceed5f7059953"],
+            });
+
+        Assert.Equal(MilvusDataType.Array, field.DataType);
+        Assert.Equal(MilvusDataType.VarChar, field.ElementType);
+        Assert.Equal(2, field.RowCount);
+        Assert.Equal(2, field.Data.Count);
+    }
 }
diff --git a/Milvus.Client/FieldData.cs b/Milvus.Client/FieldData.cs
@@ -165,7 +165,7 @@ internal static FieldData FromGrpcFieldData(Grpc.FieldData fieldData)
     internal static MilvusDataType EnsureDataType<TDataType>()
     {
         Type type = typeof(TDataType);
-        MilvusDataType dataType = MilvusDataType.Double;
+        MilvusDataType dataType = MilvusDataType.None;
 
         if (type == typeof(bool))
         {
@@ -209,7 +209,7 @@ internal static MilvusDataType EnsureDataType<TDataType>()
         }
         else
         {
-            throw new NotSupportedException($"Not Support DataType:{dataType}");
+            throw new NotSupportedException($"Type {type.Name} cannot be mapped to DataType");
         }
 
         return dataType;