diff --git a/test/test_data_table.py b/test/test_data_table.py index 721fdec88a1522a76d407b36cfa0a6c59f7c6c92..0b3343e51fca990d9a81fa700565f05958133edb 100644 --- a/test/test_data_table.py +++ b/test/test_data_table.py @@ -1,6 +1,7 @@ from __future__ import division # data from Wikipedia "join" article + def get_dept_table(): from pytools.datatable import DataTable dept_table = DataTable(["id", "name"]) @@ -10,6 +11,7 @@ def get_dept_table(): dept_table.insert_row((35, "Marketing")) return dept_table + def get_employee_table(): from pytools.datatable import DataTable employee_table = DataTable(["lastname", "dept"]) @@ -21,13 +23,15 @@ def get_employee_table(): employee_table.insert_row(("Smith", 34)) return employee_table + def test_len(): et = get_employee_table() assert len(et) == 6 + def test_iter(): et = get_employee_table() - + count = 0 for row in et: count += 1 @@ -35,20 +39,24 @@ def test_iter(): assert count == 6 + def test_insert_and_get(): et = get_employee_table() et.insert(dept=33, lastname="Kloeckner") assert et.get(lastname="Kloeckner").dept == 33 + def test_filtered(): et = get_employee_table() assert len(et.filtered(dept=33)) == 2 assert len(et.filtered(dept=34)) == 2 + def test_sort(): et = get_employee_table() et.sort(["lastname"]) - assert et.column_data("dept") == [36,33,31,34,34,33] + assert et.column_data("dept") == [36, 33, 31, 34, 34, 33] + def test_aggregate(): et = get_employee_table() @@ -60,6 +68,7 @@ def test_aggregate(): for lastname in lastnames: assert et.get(lastname=lastname).dept == dept + def test_aggregate_2(): from pytools.datatable import DataTable tbl = DataTable(["step", "value"], zip(range(20), range(20))) @@ -67,6 +76,7 @@ def test_aggregate_2(): assert agg.column_data("step") == range(20) assert agg.column_data("value") == range(20) + def test_join(): et = get_employee_table() dt = get_dept_table() @@ -83,4 +93,3 @@ def test_join(): outer_joined = et.join("dept", "id", dt, outer=True) assert len(outer_joined) == len(et)+1 -