#50 type-verbose: fixed issues.

numbworks · Dec 30, 2024 · dba01ab · dba01ab
1 parent 3cc700d
commit dba01ab
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/nwtimetrackingtests.py b/tests/nwtimetrackingtests.py
@@ -2811,8 +2811,8 @@ def test_orchestrateheadn_shouldreturntail_whenheadnisnotnoneanddisplayheadnwith
         )
 
         df : DataFrame = DataFrame({"2015": ["10h 30m", "15h 45m"], "↕": ["↑", "↑"], "2016": ["20h 45m", "20h 00m"]})
-        head_n : Optional[int] = 2
-        expected : DataFrame = df.tail(n = int(head_n))
+        head_n : Optional[uint] = uint(2)
+        expected : DataFrame = df.tail(n = cast(int, head_n))
 
         # Act
         actual : DataFrame = tt_adapter._TTAdapter__orchestrate_head_n(df = df, head_n = head_n, display_head_n_with_tail = True)    # type: ignore
@@ -2832,8 +2832,8 @@ def test_orchestrateheadn_shouldreturnhead_whenheadnisnotnoneanddisplayheadnwith
         )
 
         df : DataFrame = DataFrame({"2015": ["10h 30m", "15h 45m"], "↕": ["↑", "↑"], "2016": ["20h 45m", "20h 00m"]})
-        head_n : Optional[int] = 2
-        expected : DataFrame = df.head(n = int(head_n))
+        head_n : Optional[uint] = uint(2)
+        expected : DataFrame = df.head(n = cast(int, head_n))
 
         # Act
         actual : DataFrame = tt_adapter._TTAdapter__orchestrate_head_n(df = df, head_n = head_n, display_head_n_with_tail = False)    # type: ignore