def test_view_computed_with_row_pivots_replace(self): table = Table({"a": [1, 2, 3, 4], "b": [5, 6, 7, 8]}) view = table.view( row_pivots=["computed"], computed_columns=[{ "column": "computed", "computed_function_name": "+", "inputs": ["a", "b"], }], ) assert view.to_columns() == { "__ROW_PATH__": [[], [6], [8], [10], [12]], "a": [10, 1, 2, 3, 4], "b": [26, 5, 6, 7, 8], "computed": [36.0, 6.0, 8.0, 10.0, 12.0], } table.replace({"a": [10, 20, 30, 40], "b": [50, 60, 70, 80]}) assert view.to_columns() == { "__ROW_PATH__": [[], [60], [80], [100], [120]], "a": [100, 10, 20, 30, 40], "b": [260, 50, 60, 70, 80], "computed": [360.0, 60.0, 80.0, 100.0, 120.0], }
def test_view_computed_multiple_dependents_replace(self): table = Table({"a": [1, 2, 3, 4], "b": [5, 6, 7, 8]}) view = table.view(computed_columns=[ { "column": "computed", "computed_function_name": "+", "inputs": ["a", "b"], }, { "column": "final", "computed_function_name": "pow2", "inputs": ["computed"], }, ]) assert view.to_columns() == { "a": [1, 2, 3, 4], "b": [5, 6, 7, 8], "computed": [6, 8, 10, 12], "final": [36, 64, 100, 144], } table.replace({"a": [10, 20, 30, 40], "b": [50, 60, 70, 80]}) assert view.schema() == { "a": int, "b": int, "computed": float, "final": float, } assert view.to_columns() == { "a": [10, 20, 30, 40], "b": [50, 60, 70, 80], "computed": [60, 80, 100, 120], "final": [3600, 6400, 10000, 14400], }
def test_table_replace_views_should_preserve(self): data = [{"a": 1, "b": 2}, {"a": 3, "b": 4}] data2 = [{"a": 3, "b": 4}, {"a": 1, "b": 2}] tbl = Table(data) view = tbl.view(row_pivots=["a"], column_pivots=["b"]) first = view.to_records() tbl.replace(data2) assert view.to_records() == first
def test_table_replace_views_should_preserve(self): data = [{"a": 1, "b": 2}, {"a": 3, "b": 4}] data2 = [{"a": 3, "b": 4}, {"a": 1, "b": 2}] tbl = Table(data) view = tbl.view(group_by=["a"], split_by=["b"]) first = view.to_records() tbl.replace(data2) assert view.to_records() == first
def test_table_replace_should_fire_on_update(self, sentinel): data = [{"a": 1, "b": 2}, {"a": 3, "b": 4}] data2 = [{"a": 3, "b": 4}, {"a": 1, "b": 2}] tbl = Table(data) view = tbl.view() s = sentinel(False) def updater(port_id): assert port_id == 0 s.set(True) view.on_update(updater) tbl.replace(data2) assert s.get() is True
def test_table_replace_should_fire_on_update_with_delta(self, sentinel): data = [{"a": 1, "b": 2}, {"a": 3, "b": 4}] data2 = [{"a": 3, "b": 4}, {"a": 1, "b": 2}] tbl = Table(data) view = tbl.view() s = sentinel(False) def updater(port_id, delta): assert port_id == 0 table2 = Table(delta) assert table2.view().to_records() == data2 s.set(True) view.on_update(updater, mode="row") tbl.replace(data2) assert s.get() is True
def test_view_computed_multiple_views_should_all_replace(self): table = Table({"a": [1, 2, 3, 4], "b": [5, 6, 7, 8]}) view = table.view(computed_columns=[{ "column": "computed", "computed_function_name": "+", "inputs": ["a", "b"], }]) view2 = table.view(computed_columns=[{ "column": "computed2", "computed_function_name": "-", "inputs": ["a", "b"], }]) assert view.schema() == {"a": int, "b": int, "computed": float} assert view2.schema() == {"a": int, "b": int, "computed2": float} assert view.to_columns() == { "a": [1, 2, 3, 4], "b": [5, 6, 7, 8], "computed": [6, 8, 10, 12], } assert view2.to_columns() == { "a": [1, 2, 3, 4], "b": [5, 6, 7, 8], "computed2": [-4, -4, -4, -4], } table.replace({"a": [10, 20, 30, 40], "b": [50, 60, 70, 80]}) assert view.to_columns() == { "a": [10, 20, 30, 40], "b": [50, 60, 70, 80], "computed": [60, 80, 100, 120], } assert view2.to_columns() == { "a": [10, 20, 30, 40], "b": [50, 60, 70, 80], "computed2": [-40, -40, -40, -40], }