1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
// Licensed to the Apache Software Foundation (ASF) under one or more
// contributor license agreements. See the NOTICE file distributed with
// this work for additional information regarding copyright ownership.
// The ASF licenses this file to You under the Apache License, Version 2.0
// (the "License"); you may not use this file except in compliance with
// the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
using System;
using System.Collections.Generic;
using Apache.Arrow.Types;
using Xunit;
namespace Apache.Arrow.Tests
{
public class TableTests
{
public static Table MakeTableWithOneColumnOfTwoIntArrays(int lengthOfEachArray)
{
Array intArray = ColumnTests.MakeIntArray(lengthOfEachArray);
Array intArrayCopy = ColumnTests.MakeIntArray(lengthOfEachArray);
Field field = new Field.Builder().Name("f0").DataType(Int32Type.Default).Build();
Schema s0 = new Schema.Builder().Field(field).Build();
Column column = new Column(field, new List<Array> { intArray, intArrayCopy });
Table table = new Table(s0, new List<Column> { column });
return table;
}
[Fact]
public void TestEmptyTable()
{
Table table = new Table();
Assert.Equal(0, table.ColumnCount);
Assert.Equal(0, table.RowCount);
}
[Fact]
public void TestTableBasics()
{
Table table = MakeTableWithOneColumnOfTwoIntArrays(10);
Assert.Equal(20, table.RowCount);
Assert.Equal(1, table.ColumnCount);
}
[Fact]
public void TestTableAddRemoveAndSetColumn()
{
Table table = MakeTableWithOneColumnOfTwoIntArrays(10);
Array nonEqualLengthIntArray = ColumnTests.MakeIntArray(10);
Field field1 = new Field.Builder().Name("f1").DataType(Int32Type.Default).Build();
Column nonEqualLengthColumn = new Column(field1, new[] { nonEqualLengthIntArray});
Assert.Throws<ArgumentException>(() => table.InsertColumn(-1, nonEqualLengthColumn));
Assert.Throws<ArgumentException>(() => table.InsertColumn(1, nonEqualLengthColumn));
Array equalLengthIntArray = ColumnTests.MakeIntArray(20);
Field field2 = new Field.Builder().Name("f2").DataType(Int32Type.Default).Build();
Column equalLengthColumn = new Column(field2, new[] { equalLengthIntArray});
Column existingColumn = table.Column(0);
Table newTable = table.InsertColumn(0, equalLengthColumn);
Assert.Equal(2, newTable.ColumnCount);
Assert.True(newTable.Column(0) == equalLengthColumn);
Assert.True(newTable.Column(1) == existingColumn);
newTable = newTable.RemoveColumn(1);
Assert.Equal(1, newTable.ColumnCount);
Assert.True(newTable.Column(0) == equalLengthColumn);
newTable = table.SetColumn(0, existingColumn);
Assert.True(newTable.Column(0) == existingColumn);
}
}
}
|
