|
| 1 | +using System; |
| 2 | +using System.Collections.Generic; |
| 3 | +using System.Text; |
| 4 | +using Microsoft.Data.DataView; |
| 5 | +using Microsoft.ML.Core.Data; |
| 6 | +using Microsoft.ML.Data; |
| 7 | +using Microsoft.ML.Data.DataLoadSave; |
| 8 | +using Microsoft.ML.TestFramework; |
| 9 | +using Xunit; |
| 10 | +using Xunit.Abstractions; |
| 11 | + |
| 12 | +namespace Microsoft.ML.Tests |
| 13 | +{ |
| 14 | + public class FakeSchemaTest : BaseTestClass |
| 15 | + { |
| 16 | + public FakeSchemaTest(ITestOutputHelper output) |
| 17 | + : base(output) |
| 18 | + { |
| 19 | + } |
| 20 | + |
| 21 | + [Fact] |
| 22 | + void SimpleTest() |
| 23 | + { |
| 24 | + var metadataBuilder = new MetadataBuilder(); |
| 25 | + metadataBuilder.Add("M", NumberType.R4, (ref float v) => v = 484f); |
| 26 | + var schemaBuilder = new SchemaBuilder(); |
| 27 | + schemaBuilder.AddColumn("A", new VectorType(NumberType.R4, 94)); |
| 28 | + schemaBuilder.AddColumn("B", new KeyType(typeof(uint), 17)); |
| 29 | + schemaBuilder.AddColumn("C", NumberType.I4, metadataBuilder.GetMetadata()); |
| 30 | + |
| 31 | + var shape = SchemaShape.Create(schemaBuilder.GetSchema()); |
| 32 | + |
| 33 | + var fakeSchema = FakeSchemaFactory.Create(shape); |
| 34 | + |
| 35 | + var columnA = fakeSchema[0]; |
| 36 | + var columnB = fakeSchema[1]; |
| 37 | + var columnC = fakeSchema[2]; |
| 38 | + |
| 39 | + Assert.Equal("A", columnA.Name); |
| 40 | + Assert.Equal(NumberType.R4, columnA.Type.GetItemType()); |
| 41 | + Assert.Equal(10, columnA.Type.GetValueCount()); |
| 42 | + |
| 43 | + Assert.Equal("B", columnB.Name); |
| 44 | + Assert.Equal(DataKind.U4, columnB.Type.GetRawKind()); |
| 45 | + Assert.Equal(10u, columnB.Type.GetKeyCount()); |
| 46 | + |
| 47 | + Assert.Equal("C", columnC.Name); |
| 48 | + Assert.Equal(NumberType.I4, columnC.Type); |
| 49 | + |
| 50 | + var metaC = columnC.Metadata; |
| 51 | + Assert.Single(metaC.Schema); |
| 52 | + |
| 53 | + float mValue = -1; |
| 54 | + metaC.GetValue("M", ref mValue); |
| 55 | + Assert.Equal(default, mValue); |
| 56 | + } |
| 57 | + } |
| 58 | +} |
0 commit comments