Merge pull request #147 from dcereijodo/feature/assert-expression-on-condition

support extra parameter 'condition' on 'expression_is_true' macro

Author: drewbanin

README.md

@@ -128,6 +128,23 @@ models:
 
 ```
 
+The macro accepts an optional parameter `condition` that allows for asserting
+the `expression` on a subset of all records.
+
+Usage:
+```yaml
+version: 2
+
+models:
+  - name: model_name
+    tests:
+      - dbt_utils.expression_is_true:
+          expression: "col_a + col_b = total"
+          condition: "created_at > '2018-12-31'"
+
+```
+
+
 #### recency ([source](macros/schema_tests/recency.sql))
 This schema test asserts that there is data in the referenced model at least as recent as the defined interval prior to the current timestamp.
 

integration_tests/models/schema_tests/schema.yml

@@ -17,7 +17,10 @@ models:
     tests:
       - dbt_utils.expression_is_true:
           expression: col_a + col_b = 1
-        
+      - dbt_utils.expression_is_true:
+          expression: col_a = 0.5
+          condition: col_b = 0.5
+
   - name: test_recency
     tests:
       - dbt_utils.recency:

macros/schema_tests/expression_is_true.sql

@@ -1,12 +1,17 @@
-{% macro test_expression_is_true(model) %}
+{% macro test_expression_is_true(model, condition='true') %}
 
 {% set expression = kwargs.get('expression', kwargs.get('arg')) %}
 
-with validation_errors as (
+with meet_condition as (
+
+    select * from {{ model }} where {{ condition }}
+
+),
+validation_errors as (
 
     select
         *
-    from {{model}}
+    from meet_condition
     where not({{expression}})
 
 )