def test_addcolumn(): table1 = (('foo', 'bar'), ('A', 1), ('B', 2)) col = [True, False] expect2 = (('foo', 'bar', 'baz'), ('A', 1, True), ('B', 2, False)) table2 = addcolumn(table1, 'baz', col) ieq(expect2, table2) ieq(expect2, table2) # test short column table3 = (('foo', 'bar'), ('A', 1), ('B', 2), ('C', 2)) expect4 = (('foo', 'bar', 'baz'), ('A', 1, True), ('B', 2, False), ('C', 2, None)) table4 = addcolumn(table3, 'baz', col) ieq(expect4, table4) # test short table col = [True, False, False] expect5 = (('foo', 'bar', 'baz'), ('A', 1, True), ('B', 2, False), (None, None, False)) table5 = addcolumn(table1, 'baz', col) ieq(expect5, table5)
def test_empty_addcolumn(): table1 = empty() table2 = addcolumn(table1, 'foo', ['A', 'B']) table3 = addcolumn(table2, 'bar', [1, 2]) expect = (('foo', 'bar'), ('A', 1), ('B', 2)) ieq(expect, table3) ieq(expect, table3)