fix an assertion on a test

Author: brianjlai

unit_tests/sources/declarative/async_job/test_integration.py

@@ -38,7 +38,7 @@
 
 class MockAsyncJobRepository(AsyncJobRepository):
     def start(self, stream_slice: StreamSlice) -> AsyncJob:
-        return AsyncJob("a_job_id", StreamSlice(partition={}, cursor_slice={}))
+        return AsyncJob("a_job_id", stream_slice)
 
     def update_jobs_status(self, jobs: Set[AsyncJob]) -> None:
         for job in jobs:

unit_tests/sources/declarative/partition_routers/test_async_job_partition_router.py

@@ -78,5 +78,7 @@ def test_stream_slices_with_parent_slicer():
         attempts_per_job = list(partition.jobs)
         assert len(attempts_per_job) == 1
         assert attempts_per_job[0].api_job_id() == "a_job_id"
-        assert attempts_per_job[0].job_parameters() == StreamSlice(partition={}, cursor_slice={})
+        assert attempts_per_job[0].job_parameters() == StreamSlice(
+            partition={"parent_id": str(i)}, cursor_slice={}
+        )
         assert attempts_per_job[0].status() == AsyncJobStatus.COMPLETED