Handle missing batch_start

dbt-labs · Jan 17, 2025 · 65a1db0 · 65a1db0
1 parent fe73096
commit 65a1db0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/unit/materializations/incremental/test_microbatch.py b/tests/unit/materializations/incremental/test_microbatch.py
@@ -605,6 +605,7 @@ def test_truncate_timestamp(self, timestamp, batch_size, expected_timestamp):
     @pytest.mark.parametrize(
         "batch_size,start_time,expected_formatted_start_time",
         [
+            (BatchSize.year, None, ""),
             (BatchSize.year, datetime(2020, 1, 1, 1), "20200101"),
             (BatchSize.month, datetime(2020, 1, 1, 1), "20200101"),
             (BatchSize.day, datetime(2020, 1, 1, 1), "20200101"),
@@ -617,6 +618,7 @@ def test_batch_id(self, batch_size, start_time, expected_formatted_start_time):
     @pytest.mark.parametrize(
         "batch_size,batch_start,expected_formatted_batch_start",
         [
+            (BatchSize.year, None, ""),
             (BatchSize.year, datetime(2020, 1, 1, 1), "20200101"),
             (BatchSize.month, datetime(2020, 1, 1, 1), "20200101"),
             (BatchSize.day, datetime(2020, 1, 1, 1), "20200101"),