Wrapper for Great Expectations to fit the requirements of the Gemeente Amsterdam.
Project description
Introduction
This repository contains functions that will ease the use of Great Expectations. Users can input data and data quality rules and get results in return.
DISCLAIMER: The package is in MVP phase
Getting started
Install the dq suite on your compute, for example by running the following code in your workspace:
pip install dq-suite-amsterdam
import dq_suite
Load your data in dataframes, give them a table_name, and create a list of all dataframes:
df = spark.read.csv(csv_path+file_name, header=True, inferSchema=True) #example using csv
df.table_name = "showcase_table"
dfs = [df]
- Define 'dfs' as a list of dataframes that require a dq check
- Define 'dq_rules' as a JSON as shown in dq_rules_example.json in this repo
- Define a name for your dq check, in this case "showcase"
dq_suite.df_check(dfs, dq_rules, "dpxx_dev", "showcase", spark)
Create dataquality schema and tables (in respective catalog of data team)
for the first time installation create data quality schema and tables from the notebook from repo path scripts/data_quality_tables.sql
- open the notebook, connect to a cluster
- select the catalog of the data team and execute the notebook. It will check if schema is available if not it will create schema and same for tables.
Export the schema from Unity Catalog to the Input Form
In order to output the schema from Unity Catalog, use the following commands (using the required schema name):
schema_output = dq_suite.export_schema('schema_name', spark)
print(schema_output)
Copy the string to the Input Form to quickly ingest the schema in Excel.
Validate the schema of a table
It is possible to validate the schema of an entire table to a schema definition from Amsterdam Schema in one go. This is done by adding two fields to the "dq_rules" JSON when describing the table (See: https://github.com/Amsterdam/dq-suite-amsterdam/blob/main/dq_rules_example.json).
You will need:
- validate_table_schema: the id field of the table from Amsterdam Schema
- validate_table_schema_url: the url of the table or dataset from Amsterdam Schema
The schema definition is converted into column level expectations (expect_column_values_to_be_of_type) on run time.
Known exceptions
The functions can run on Databricks using a Personal Compute Cluster or using a Job Cluster. Using a Shared Compute Cluster will results in an error, as it does not have the permissions that Great Expectations requires.
Updates
Version 0.1: Run a DQ check for a dataframe
Version 0.2: Run a DQ check for multiple dataframes
Version 0.3: Refactored I/O
Version 0.4: Added schema validation with Amsterdam Schema per table
Version 0.5: Export schema from Unity Catalog
Version 0.6: The results are written to tables in the "dataquality" schema
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
File details
Details for the file dq_suite_amsterdam-0.6.1.tar.gz.
File metadata
- Download URL: dq_suite_amsterdam-0.6.1.tar.gz
- Upload date:
- Size: 8.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.0 CPython/3.12.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 4f5bf5be4a189b852a6bcd54b0f7c0f95caf9a4ffb0908a505af0746f329047b |
|
| MD5 | b420b916ca69157021b5fc2a4e380448 |
|
| BLAKE2b-256 | 194ad8f1519d1226269c327013269be766791502cebf5f6477dee79bac85c7ad |
File details
Details for the file dq_suite_amsterdam-0.6.1-py3-none-any.whl.
File metadata
- Download URL: dq_suite_amsterdam-0.6.1-py3-none-any.whl
- Upload date:
- Size: 8.2 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.1.0 CPython/3.12.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 82372a0655a27463f4864aeda4b55b7b10bbcafd03e1e49e8b7776190e5a9338 |
|
| MD5 | 165ce0999578d85404eaed5aba490cbb |
|
| BLAKE2b-256 | 4c2783f57511ac1c8a3687952cf47e5c5629a738a0c7bf1ae854839465adf313 |
