Fix failing test.

enso-org · Jan 17, 2024 · 5b2af6e · 5b2af6e
1 parent 746888f
commit 5b2af6e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/test/Table_Tests/src/In_Memory/Table_Spec.enso b/test/Table_Tests/src/In_Memory/Table_Spec.enso
@@ -898,8 +898,12 @@ spec =
             t_range.at "Range" . to_vector . should_equal [100, 101, 102]
 
             t_date_range = t.set ((Date.new 2020 1 1).up_to (Date.new 2020 1 4))
-            t_date_range.column_names.should_equal ["X", "Date Range"]
-            t_date_range.at "Date Range" . to_vector . should_equal [Date.new 2020 1 1, Date.new 2020 1 2, Date.new 2020 1 3]
+            t_date_range.column_names.should_equal ["X", "Date"]
+            t_date_range.at "Date" . to_vector . should_equal [Date.new 2020 1 1, Date.new 2020 1 2, Date.new 2020 1 3]
+
+            t_month_range = t.set ((Date.new 2020 1 1).up_to (Date.new 2020 4 1) step=Date_Period.Month)
+            t_month_range.column_names.should_equal ["X", "month"]
+            t_month_range.at "Month" . to_vector . should_equal [Date.new 2020 1 1, Date.new 2020 2 1, Date.new 2020 3 1]
 
         Test.specify "should fail if there is a length mismatch on a new column" <|
             t = Table.new [["X", [1, 2, 3]]]