From d5ef61bfcaee9e68df28f060d1849fd3a2a05678 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Fri, 22 Nov 2013 10:44:08 -0600 Subject: [PATCH] PEP8 datatable test --- test/test_data_table.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/test/test_data_table.py b/test/test_data_table.py index 721fdec..0b3343e 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 - -- GitLab