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_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 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