@@ -198,7 +198,7 @@ def test_add_interval(snapshot: Snapshot, make_snapshot):
198198 ]
199199
200200
201- def test_add_interval_dev (snapshot : Snapshot ):
201+ def test_add_interval_dev (snapshot : Snapshot , make_snapshot ):
202202 snapshot .version = "existing_version"
203203 snapshot .change_category = SnapshotChangeCategory .FORWARD_ONLY
204204
@@ -209,6 +209,18 @@ def test_add_interval_dev(snapshot: Snapshot):
209209 assert snapshot .intervals == [(to_timestamp ("2020-01-01" ), to_timestamp ("2020-01-02" ))]
210210 assert snapshot .dev_intervals == [(to_timestamp ("2020-01-02" ), to_timestamp ("2020-01-03" ))]
211211
212+ new_snapshot = make_snapshot (snapshot .model )
213+ new_snapshot .merge_intervals (snapshot )
214+ assert new_snapshot .intervals == [(to_timestamp ("2020-01-01" ), to_timestamp ("2020-01-02" ))]
215+ assert new_snapshot .dev_intervals == []
216+
217+ new_snapshot = make_snapshot (snapshot .model )
218+ new_snapshot .previous_versions = snapshot .all_versions
219+ new_snapshot .migrated = True
220+ new_snapshot .merge_intervals (snapshot )
221+ assert new_snapshot .intervals == [(to_timestamp ("2020-01-01" ), to_timestamp ("2020-01-02" ))]
222+ assert new_snapshot .dev_intervals == [(to_timestamp ("2020-01-02" ), to_timestamp ("2020-01-03" ))]
223+
212224
213225def test_add_interval_partial (snapshot : Snapshot , make_snapshot ):
214226 snapshot .add_interval ("2023-01-01 00:00:00" , "2023-01-01 23:59:59" )
0 commit comments