Fix ChainedAssignmentError warnings in test code

red-data-tools · May 17, 2024 · 61a9779 · 61a9779
1 parent a8fb0ff
commit 61a9779
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 14 deletions.
diff --git a/test/plot_methods/hist_plot_test.rb b/test/plot_methods/hist_plot_test.rb
@@ -44,10 +44,12 @@ def setup_arrow_data
 
       def setup_pandas_data
         pandas_required
-        @data = Pandas::DataFrame.new(data: @array_data)
-        @data[:red] = @data[:red].astype("float64")
-        @data[:blue] = @data[:blue].astype("float64")
-        @data[:green] = @data[:green].astype("float64")
+        data = Pandas::DataFrame.new(data: @array_data)
+        @data = Pandas::DataFrame.new(data: {
+          red: data[:red].astype("float64"),
+          blue: data[:blue].astype("float64"),
+          green: data[:green].astype("float64")
+        })
       end
     end
 
@@ -86,9 +88,11 @@ def setup_arrow_data
 
     def setup_pandas_data
       pandas_required
-      @data = Pandas::DataFrame.new(data: @array_data)
-      @data[:a] = @data[:a].astype("float64")
-      @data[:c] = @data[:c].astype("category")
+      data = Pandas::DataFrame.new(data: @array_data)
+      @data = Pandas::DataFrame.new(data: {
+        a: data[:a].astype("float64"),
+        c: data[:c].astype("category")
+      })
     end
   end
 end
diff --git a/test/table/table_group_by_test.rb b/test/table/table_group_by_test.rb
@@ -237,9 +237,9 @@ def setup_table_by_pandas
     @expected_applied_table.adapter.data.each do |key, value|
       data[key] = value.to_a
     end
+    data[:a_min] = data[:a_min].map(&:to_f)
+    data[:a_max] = data[:a_max].map(&:to_f)
     df = Pandas::DataFrame.new(data: data)
-    df[:a_min] = df[:a_min].astype(:float64)
-    df[:a_max] = df[:a_max].astype(:float64)
     @expected_applied_table = Charty::Table.new(df, index: @expected_applied_table.index)
   end
 

diff --git a/test/table/table_reset_index_test.rb b/test/table/table_reset_index_test.rb
@@ -105,9 +105,12 @@ def setup_table_by_pandas
 
       @table = Charty::Table.new(Pandas::DataFrame.new(data: @data))
 
-      df = Pandas::DataFrame.new(data: {b: @expected_indices.keys}.merge(@expected_applied_data))
-      df[:a_min] = df[:a_min].astype(:float64)
-      df[:a_max] = df[:a_max].astype(:float64)
+      df = Pandas::DataFrame.new(data: {
+        b: @expected_indices.keys,
+        a: @expected_applied_data[:a],
+        a_min: @expected_applied_data[:a_min].map(&:to_f),
+        a_max: @expected_applied_data[:a_max].map(&:to_f)
+      })
       @expected_result = Charty::Table.new(df)
     end
   end

diff --git a/test/vector/pandas_test.rb b/test/vector/pandas_test.rb
@@ -71,10 +71,10 @@ def test_aref
                        5
                      ],
                      [
-                       @vector[1],
+                       @vector.iloc(1),
                        @vector["c"],
                        @vector["d"],
-                       @vector[4]
+                       @vector.iloc(4)
                      ])
       end
     end