Test: Fix pandas expected data in table_aset_test.rb

red-data-tools · May 20, 2024 · 1d8571c · 1d8571c
1 parent e10f6eb
commit 1d8571c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/table/table_aset_test.rb b/test/table/table_aset_test.rb
@@ -131,8 +131,8 @@ def test_aset_existing(data)
             end
       table[key] = [10, 20, 30, 40, 50]
       expected_data = {
-        foo: [1, 2, 3, 4, 5],
-        bar: [10, 20, 30, 40, 50]
+        foo: Pandas::Series.new([1, 2, 3, 4, 5]),
+        bar: Pandas::Series.new([10, 20, 30, 40, 50], dtype: "object")
       }
       expected_data = transform_keys(expected_data, &:to_s) if data_key_type == :string
       expected_data = Pandas::DataFrame.new(data: expected_data)
@@ -152,8 +152,8 @@ def test_aset_new(data)
             end
       table[key] = [10, 20, 30, 40, 50]
       expected_data = {
-        foo: [1, 2, 3, 4, 5],
-        bar: [10, 20, 30, 40, 50]
+        foo: Pandas::Series.new([1, 2, 3, 4, 5]),
+        bar: Pandas::Series.new([10, 20, 30, 40, 50], dtype: "object")
       }
       expected_data = transform_keys(expected_data, &:to_s) if data_key_type == :string
       expected_data = Pandas::DataFrame.new(data: expected_data)