def test_aggregate_2():
from pytools.datatable import DataTable
tbl = DataTable(["step", "value"],
list(zip(list(range(20)), list(range(20)))))
agg = tbl.aggregated(["step"], "value", max)
assert agg.column_data("step") == list(range(20))
assert agg.column_data("value") == list(range(20))
def get_dept_table():
from pytools.datatable import DataTable
dept_table = DataTable(["id", "name"])
dept_table.insert_row((31, "Sales"))
dept_table.insert_row((33, "Engineering"))
dept_table.insert_row((34, "Clerical"))
dept_table.insert_row((35, "Marketing"))
return dept_table
def get_warnings(self):
from pytools.datatable import DataTable
result = DataTable(["step", "message", "category", "filename", "lineno"])
if self.schema_version >= 1 and self.db_conn is not None:
for row in self.db_conn.execute(
"select step, message, category, filename, lineno from warnings"):
result.insert_row(row)
return result
def get_dept_table():
from pytools.datatable import DataTable
dept_table = DataTable(["id", "name"])
dept_table.insert_row((31, "Sales"))
dept_table.insert_row((33, "Engineering"))
dept_table.insert_row((34, "Clerical"))
dept_table.insert_row((35, "Marketing"))
return dept_table
def get_table(self, q_name):
if q_name not in self.quantity_data:
raise KeyError("invalid quantity name '%s'" % q_name)
from pytools.datatable import DataTable
result = DataTable(["step", "rank", "value"])
for row in self.db_conn.execute(
"select step, rank, value from %s" % q_name):
result.insert_row(row)
return result
def get_warnings(self):
columns = ["step", "message", "category", "filename", "lineno"]
if self.schema_version >= 2:
columns.insert(0, "rank")
from pytools.datatable import DataTable
result = DataTable(columns)
for row in self.db_conn.execute(
"select %s from warnings" % (", ".join(columns))):
result.insert_row(row)
return result
def test_aggregate_2():
from pytools.datatable import DataTable
tbl = DataTable(["step", "value"], list(zip(list(range(20)), list(range(20)))))
agg = tbl.aggregated(["step"], "value", max)
assert agg.column_data("step") == list(range(20))
assert agg.column_data("value") == list(range(20))
def get_employee_table():
from pytools.datatable import DataTable
employee_table = DataTable(["lastname", "dept"])
employee_table.insert_row(("Rafferty", 31))
employee_table.insert_row(("Jones", 33))
employee_table.insert_row(("Jasper", 36))
employee_table.insert_row(("Steinberg", 33))
employee_table.insert_row(("Robinson", 34))
employee_table.insert_row(("Smith", 34))
return employee_table
def get_employee_table():
from pytools.datatable import DataTable
employee_table = DataTable(["lastname", "dept"])
employee_table.insert_row(("Rafferty", 31))
employee_table.insert_row(("Jones", 33))
employee_table.insert_row(("Jasper", 36))
employee_table.insert_row(("Steinberg", 33))
employee_table.insert_row(("Robinson", 34))
employee_table.insert_row(("Smith", 34))
return employee_table
