예제 #1
0
    def testSplitBy(self):
        t = toTable("a", [1, 1, 3, 4])
        t.addColumn("b", [1, 1, 3, 3])
        t.addColumn("c", [1, 2, 1, 4])
        t._print()
        subts = t.splitBy("a")
        assert len(subts) == 3
        res = Table.mergeTables(subts)
        assert len(res) == len(t)
        subts[0]._print()
        assert res.a.values == t.a.values
        assert res.b.values == t.b.values
        assert res.c.values == t.c.values

        # check if input tables are not altered
        for subt in subts:
            assert subt.getColNames() == ["a", "b", "c"]

        subts = t.splitBy("a", "c")
        assert len(subts) == 4
        res = Table.mergeTables(subts)
        assert res.a.values == t.a.values
        assert res.b.values == t.b.values
        assert res.c.values == t.c.values

        # check if input tables are not altered
        for subt in subts:
            assert subt.getColNames() == ["a", "b", "c"]
예제 #2
0
파일: test_table.py 프로젝트: lowks/emzed2
    def testSplitBy(self):
        t = toTable("a", [1,1,3,4])
        t.addColumn("b", [1,1,3,3])
        t.addColumn("c", [1,2,1,4])
        t._print()
        subts = t.splitBy("a")
        assert len(subts) == 3
        res = Table.mergeTables(subts)
        assert len(res) == len(t)
        subts[0]._print()
        assert res.a.values == t.a.values
        assert res.b.values == t.b.values
        assert res.c.values == t.c.values

        # check if input tables are not altered
        for subt in subts:
            assert subt.getColNames() == [ "a", "b", "c"]

        subts = t.splitBy("a", "c")
        assert len(subts) == 4
        res = Table.mergeTables(subts)
        assert res.a.values == t.a.values
        assert res.b.values == t.b.values
        assert res.c.values == t.c.values

        # check if input tables are not altered
        for subt in subts:
            assert subt.getColNames() == [ "a", "b", "c"]
예제 #3
0
    def testMerge(self):
        t1 = toTable("a", [1])
        t1.addColumn("b", [2])
        t1.addColumn("c", [3])

        t2 = toTable("a", [1, 2])
        t2.addColumn("c", [1, 3])
        t2.addColumn("d", [1, 4])

        tn = Table.mergeTables([t1, t2])

        assert tn.a.values == (
            1,
            1,
            2,
        )
        assert tn.b.values == (
            2,
            None,
            None,
        )
        assert tn.c.values == (
            3,
            1,
            3,
        )
        assert tn.d.values == (
            None,
            1,
            4,
        )

        # check if input tables are not altered
        assert t1.getColNames() == ["a", "b", "c"]
        assert t2.getColNames() == ["a", "c", "d"]
예제 #4
0
파일: test_table.py 프로젝트: lowks/emzed2
    def testMerge(self):
        t1 = toTable("a", [1])
        t1.addColumn("b", [2])
        t1.addColumn("c", [3])

        t2 = toTable("a", [1,2])
        t2.addColumn("c", [1,3])
        t2.addColumn("d", [1,4])

        tn = Table.mergeTables([t1, t2])

        assert tn.a.values == (1, 1, 2,)
        assert tn.b.values == (2, None, None,)
        assert tn.c.values == (3, 1, 3,)
        assert tn.d.values == (None, 1, 4,)

        # check if input tables are not altered
        assert t1.getColNames() == [ "a", "b", "c"]
        assert t2.getColNames() == [ "a", "c", "d"]