docs: point to patient object tests

Aarhus-Psychiatry-Research · Aug 25, 2023 · d727664 · d727664
1 parent 864f59b
commit d727664
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/psycop/common/feature_generation/sequences/test_cohort_definer_to_patients.py b/psycop/common/feature_generation/sequences/test_cohort_definer_to_patients.py
@@ -28,7 +28,7 @@ def get_filtered_prediction_times_bundle() -> FilteredPredictionTimeBundle:
     def get_outcome_timestamps() -> pl.DataFrame:
         df = str_to_pl_df(
             """dw_ek_borger,timestamp
-    1,2021-01-01
+    1,2021-01-02
     """,
         )
         return df
@@ -50,3 +50,4 @@ def test_polars_dataframe_to_dict():
     assert len(prediction_times) == 2
     patient_1 = list(filter(lambda x: x.patient.patient_id == 1, prediction_times))[0]
     assert patient_1.prediction_timestamp == dt.datetime(2021, 1, 1)
+    # The rest of the prediction time creation logic is tested in the patient object tests