Spaces:
Sleeping
Sleeping
| """Tests for operations module.""" | |
| import pytest | |
| from visual_notebook.domain.operations import OPERATIONS | |
| def test_operations_registry(): | |
| assert "filter" in OPERATIONS | |
| assert "sort" in OPERATIONS | |
| assert "select" in OPERATIONS | |
| assert "rename" in OPERATIONS | |
| assert "groupby" in OPERATIONS | |
| assert "fill_nulls" in OPERATIONS | |
| assert "clip" in OPERATIONS | |
| def test_filter_validation(): | |
| op = OPERATIONS["filter"] | |
| with pytest.raises(ValueError): | |
| op.validate({}, []) | |
| with pytest.raises(ValueError): | |
| op.validate({"column": "x"}, [{"name": "y", "dtype": "VARCHAR"}]) | |
| result = op.validate({"column": "x", "operator": "eq", "value": 1}, [{"name": "x", "dtype": "INTEGER"}]) | |
| assert result["column"] == "x" | |
| def test_sort_validation(): | |
| op = OPERATIONS["sort"] | |
| result = op.validate({"column": "x"}, []) | |
| assert result["descending"] is False | |
| def test_select_validation(): | |
| op = OPERATIONS["select"] | |
| with pytest.raises(ValueError): | |
| op.validate({}, []) | |
| result = op.validate({"columns": ["a", "b"]}, []) | |
| assert result["columns"] == ["a", "b"] | |