feat: add _with_context variants for generic tests

[joostboon]

Summary

• Adds 6 new elementary generic tests that extend common dbt / dbt_utils / dbt_expectations tests with a context_columns parameter
• When context_columns is provided, only those columns (plus the tested column) are returned alongside failing rows
• When context_columns is omitted, all columns are returned (SELECT *)
• Non-existent columns in context_columns emit a WARNING log and are skipped rather than raising an error

New tests

Test	Extends
elementary.not_null_with_context	dbt.not_null
elementary.accepted_range_with_context	dbt_utils.accepted_range
elementary.expect_column_values_to_not_be_null_with_context	dbt_expectations.expect_column_values_to_not_be_null
elementary.expect_column_values_to_be_unique_with_context	dbt_expectations.expect_column_values_to_be_unique
elementary.expect_column_values_to_match_regex_with_context	dbt_expectations.expect_column_values_to_match_regex (requires dbt_expectations installed)
elementary.relationships_with_context	dbt.relationships

Usage

columns:
  - name: order_id
    tests:
      - elementary.not_null_with_context:
          context_columns: [customer_id, order_date]  # omit to return all columns

      - elementary.relationships_with_context:
          to: ref('customers')
          field: id
          context_columns: [customer_id, amount]

      - elementary.accepted_range_with_context:
          min_value: 0
          max_value: 1000
          context_columns: [customer_id, order_date]

Test plan

• Run elementary.not_null_with_context on a column with nulls — verify failing rows include the specified context columns
• Run without context_columns — verify all columns are returned
• Specify a non-existent column in context_columns — verify a WARNING is logged and the column is skipped without erroring
• Run elementary.relationships_with_context on a column with broken FK references — verify context columns are correctly prefixed with the child alias
• Run elementary.expect_column_values_to_be_unique_with_context — verify all individual duplicate rows are returned (not just the grouped count)

🤖 Generated with Claude Code

Summary by CodeRabbit

• New Features
  • Added six test variants with optional context-column support: accepted_range_with_context, not_null_with_context, expect_column_values_to_not_be_null_with_context, expect_column_values_to_be_unique_with_context, expect_column_values_to_match_regex_with_context, relationships_with_context. Failing-row outputs can include selected context columns (defaults to all columns when not specified) and respect optional row filters and range/regex options.
• Chores
  • Updated common test configuration mapping to include entries for the new tests.

---

[github-actions]

👋 @joostboon
Thank you for raising your pull request.
Please make sure to add tests and document all user-facing changes.
You can do this by editing the docs files in the elementary repository.

[coderabbitai]

📝 Walkthrough

Walkthrough

Adds six new dbt generic test macros that accept optional context_columns to include additional model columns in failing-row output; updates the common test configuration mapping to register these context-aware tests.

Changes

Cohort / File(s)	Summary
New context-aware test macros macros/edr/tests/test_not_null_with_context.sql, macros/edr/tests/test_accepted_range_with_context.sql, macros/edr/tests/test_expect_column_values_to_not_be_null_with_context.sql, macros/edr/tests/test_expect_column_values_to_be_unique_with_context.sql, macros/edr/tests/test_expect_column_values_to_match_regex_with_context.sql, macros/edr/tests/test_relationships_with_context.sql	Added six test macros that accept an optional context_columns parameter. Each macro validates requested context columns against the model (case-insensitive), logs warnings for missing columns, skips duplicates of the target column, and falls back to SELECT * when context_columns is not supplied or invalid. Core test predicates remain consistent with existing expectations.
Test configuration registry macros/utils/common_test_configs.sql	Extended the common_tests_configs_mapping to register the six new context-aware tests, each with quality_dimension, failed_row_count_calc: count(*), and descriptions noting inclusion of additional context columns (defaults to all columns when not provided).

Estimated code review effort

🎯 3 (Moderate) | ⏱️ ~20 minutes

Poem

🐇 I hop through tables, sniff a missing bit,
I call my context friends to help out of it,
If names don't match, I whisper a small chide,
Then show the rows where errors like to hide,
A rabbit's cheer for tests that now provide.

🚥 Pre-merge checks | ✅ 3

✅ Passed checks (3 passed)

Check name	Status	Explanation
Title check	✅ Passed	The title 'feat: add _with_context variants for generic tests' accurately describes the main change: adding new generic test variants with context column support across six dbt tests.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.

_{✏️ Tip: You can configure your own custom pre-merge checks in the settings.}

✨ Finishing Touches

📝 Generate docstrings

• Create stacked PR
• Commit on current branch

🧪 Generate unit tests (beta)

• Create PR with unit tests
• Commit unit tests in branch feat/add-with-context-tests

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands and usage tips.}

[devin-ai-integration]

✅ Devin Review: No Issues Found

Devin Review analyzed this PR and found no potential bugs to report.

View in Devin Review to see 4 additional findings.