def test_normalize() -> None: data_table = DataTable({"a": [1, 2, 3], "b": [3, 4], "c": []}) data_table.normalize() assert data_table == { "a": [1, 2, 3], "b": [3, 4, data_table.NOT_SET], "c": [data_table.NOT_SET, data_table.NOT_SET, data_table.NOT_SET], } assert data_table.get_record(0) == {"a": 1, "b": 3, "c": None} data_table = DataTable({"a": [1, 2], "b": [3, 4]}) data_table.normalize() assert data_table == {"a": [1, 2], "b": [3, 4]}
def test_get_set_column_names() -> None: data_table = DataTable({"a": [1], "b": [DataTable.NOT_SET], "c": []}) assert data_table.get_set_column_names() == ["a", "c"] data_table.normalize() assert data_table.get_set_column_names() == ["a"]