Fix tests

Author: treysp

tests/core/test_integration.py

@@ -1867,13 +1867,16 @@ def test_snapshot_triggers(init_and_plan_context: t.Callable, mocker: MockerFixt
         for k, v in actual_triggers.items()
         if v.restatement_triggers
     }
-    assert actual_triggers_name == {
-        waiter_revenue_by_day_name: [waiter_revenue_by_day_name, order_items_name],
-        order_items_name: [order_items_name],
-        '"memory"."sushi"."top_waiters"': [waiter_revenue_by_day_name],
-        '"memory"."sushi"."customer_revenue_by_day"': [order_items_name],
-        '"memory"."sushi"."customer_revenue_lifetime"': [order_items_name],
-    }
+
+    assert sorted(actual_triggers_name[waiter_revenue_by_day_name]) == sorted(
+        [waiter_revenue_by_day_name, order_items_name]
+    )
+    assert actual_triggers_name[order_items_name] == [order_items_name]
+    assert actual_triggers_name['"memory"."sushi"."top_waiters"'] == [waiter_revenue_by_day_name]
+    assert actual_triggers_name['"memory"."sushi"."customer_revenue_by_day"'] == [order_items_name]
+    assert actual_triggers_name['"memory"."sushi"."customer_revenue_lifetime"'] == [
+        order_items_name
+    ]
 
     # RUN: select and auto-restatement triggers
     # User selects top_waiters and waiter_revenue_by_day, others added as auto-upstream

tests/core/test_scheduler.py

@@ -56,7 +56,10 @@ def test_interval_params(scheduler: Scheduler, sushi_context_fixed_date: Context
     start_ds = "2022-01-01"
     end_ds = "2022-02-05"
 
-    assert compute_interval_params([orders, waiter_revenue], start=start_ds, end=end_ds) == {
+    interval_params, _ = compute_interval_params(
+        [orders, waiter_revenue], start=start_ds, end=end_ds
+    )
+    assert interval_params == {
         orders: [
             (to_timestamp(start_ds), to_timestamp("2022-02-06")),
         ],
@@ -88,7 +91,8 @@ def test_interval_params_nonconsecutive(scheduler: Scheduler, orders: Snapshot):
 
     orders.add_interval("2022-01-10", "2022-01-15")
 
-    assert compute_interval_params([orders], start=start_ds, end=end_ds) == {
+    interval_params, _ = compute_interval_params([orders], start=start_ds, end=end_ds)
+    assert interval_params == {
         orders: [
             (to_timestamp(start_ds), to_timestamp("2022-01-10")),
             (to_timestamp("2022-01-16"), to_timestamp("2022-02-06")),