在JAVA实现DataTable对象（四）——单元测试

由于好久没来这里了，发现有几个朋友想要代码参考一下，所以上传到这里，感兴趣的朋友可以下来参考 

https://files.cnblogs.com/hjzhang/data.rar

 

在实现DataTable的基本功能后，我们对我们实现的对象进行一下测试，测试代码如下：

import java.util.List;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;


/**
     * 此类描述的是：  测试数据表对象DataTable对象相关功能是否正常
     * @author: James Cheung
     * @version: 2.0
     */
public class DataTableTest {

    @Before
    public void init() {

    }

    @After
    public void dispose() {

    }


    @Test
    public void tableInitTest() {
        DataTable table = new DataTable(); //新建一个数据表对象，不带构造函数参数
        fillTable(table);//填充数据
        Assert.assertNotNull(table);
        Assert.assertEquals(20, table.getColumns().size());
        Assert.assertEquals(100, table.getRows().size());
        //printTable(table);
        System.out.print("------------------------------------------------------------------------------------------------------------------------\n");
        table = new DataTable("TestTable");//新建一个数据表对象，带表名为参数
        //测试添加数据列
        fillTable(table);//填充数据
        Assert.assertNotNull(table);
        //printTable(table);
    }

    @Test
    public void tableSelectTest() throws Exception {
        //测试数据表的选择过滤方法
        DataTable table = new DataTable(); //新建一个数据表对象，不带构造函数参数
        fillTable(table);//填充数据
        List<DataRow> rows = table.select("col_1> 2 && col_2<10");
        Assert.assertNotNull(rows);
        Assert.assertEquals(2, rows.size());
        //测试查询指定列
        DataTable filterResult = table.select("col_1>2 && col_2<10",
                "col_1,col_2,col_3,col_4".split(","),
                true);
        printTable(filterResult);

    }

    private void printTable(DataTable table) {

        for (int i = 0; i < table.getTotalCount(); i++) {
            for (int j = 0; j < table.getColumns().size(); j++) {
                System.out.print(table.getValue(i, j));
                if (j < 19)
                    System.out.print(" , ");
            }
            System.out.print("\n");
        }
    }

    private void fillTable(DataTable table) {
        try {
            //填充20列数据
            for (int i = 0; i < 20; i++) {
                table.addColumn("Col_" + i, i % 13);
            }
            //填充100行数据
            for (int i = 0; i < 100; i++) {
                DataRow row = table.newRow();
                for (int j = 0; j < 20; j++) {
                    //循环每列加入数据
                    row.setValue(j, i * j);
                }
                table.addRow(row);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

后记：仔细看的朋友可能发现了，DataTable里还有几个函数没有实现，那几个函数是我们后面要实现的基于DataTable进行分组汇总计算所需的功能，有兴趣的朋友也可以自己来实现一套试试，我这里可以提个醒给大家，下面是类似功能的测试代码：

  newTable = dataTable.Select("tableName", "count(Name),sum(11) aaa,sum(222) name,avg(aa),last,first,max,min aa,Name", "", "Name");