Added flags method to DataFrame

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 - Updated Polars to 0.40.0
 - Added `date_ranges` method to `Polars`
+- Added `flags` method to `DataFrame`
 - Added support for keyword arguments to `agg` methods
 - Aliased `apply` to `map_rows` for `DataFrame`
 - Changed default `name` for `with_row_index` from `row_nr` to `index`

lib/polars/data_frame.rb

@@ -154,6 +154,13 @@ def dtypes
       _df.dtypes
     end
 
+    # Get flags that are set on the columns of this DataFrame.
+    #
+    # @return [Hash]
+    def flags
+      self.columns.to_h { |name| [name, self[name].flags] }
+    end
+
     # Get the schema.
     #
     # @return [Hash]

test/data_frame_test.rb

@@ -140,6 +140,11 @@ def test_dtypes
     assert_equal [Polars::Int64, Polars::String], df.dtypes
   end
 
+  def test_flags
+    df = Polars::DataFrame.new({"a" => [1, 2, 3], "b" => ["one", "two", "three"]})
+    assert df.flags
+  end
+
   def test_schema
     df = Polars::DataFrame.new({"a" => [1, 2, 3], "b" => ["one", "two", "three"]})
     expected = {"a" => Polars::Int64, "b" => Polars::String}