Contents

module Polars
  module Utils
    def self._process_null_values(null_values)
      if null_values.is_a?(Hash)
        null_values.to_a
      else
        null_values
      end
    end

    def self._is_iterable_of(val, eltype)
      val.all? { |x| x.is_a?(eltype) }
    end

    def self.is_path_or_str_sequence(val)
      val.is_a?(::Array) && val.all? { |x| pathlike?(x) }
    end

    def self.is_bool_sequence(val)
      val.is_a?(::Array) && val.all? { |x| x == true || x == false }
    end

    def self.is_int_sequence(val)
      val.is_a?(::Array) && _is_iterable_of(val, Integer)
    end

    def self.is_str_sequence(val, allow_str: false)
      if allow_str == false && val.is_a?(::String)
        false
      else
        val.is_a?(::Array) && _is_iterable_of(val, ::String)
      end
    end

    def self.arrlen(obj)
      if obj.is_a?(Range)
        # size only works for numeric ranges
        obj.to_a.length
      elsif obj.is_a?(::String)
        nil
      else
        obj.length
      end
    rescue
      nil
    end

    def self.normalize_filepath(path, check_not_directory: true)
      path = File.expand_path(path) if !path.is_a?(::String) || path.start_with?("~")
      if check_not_directory && File.exist?(path) && Dir.exist?(path)
        raise ArgumentError, "Expected a file path; #{path} is a directory"
      end
      path
    end

    def self.scale_bytes(sz, to:)
      scaling_factor = {
        "b" => 1,
        "k" => 1024,
        "m" => 1024 ** 2,
        "g" => 1024 ** 3,
        "t" => 1024 ** 4
      }[to[0]]
      if scaling_factor > 1
        sz / scaling_factor.to_f
      else
        sz
      end
    end

    def self.extend_bool(value, n_match, value_name, match_name)
      values = bool?(value) ? [value] * n_match : value
      if n_match != values.length
        msg = "the length of `#{value_name}` (#{values.length}) does not match the length of `#{match_name}` (#{n_match})"
        raise ValueError, msg
      end
      values
    end
  end
end

Version data entries

8 entries across 8 versions & 1 rubygems

Version	Path
polars-df-0.15.0-x86_64-linux	lib/polars/utils/various.rb
polars-df-0.15.0-x86_64-linux-musl	lib/polars/utils/various.rb
polars-df-0.15.0-x86_64-darwin	lib/polars/utils/various.rb
polars-df-0.15.0-x64-mingw-ucrt	lib/polars/utils/various.rb
polars-df-0.15.0-arm64-darwin	lib/polars/utils/various.rb
polars-df-0.15.0-aarch64-linux	lib/polars/utils/various.rb
polars-df-0.15.0-aarch64-linux-musl	lib/polars/utils/various.rb
polars-df-0.15.0	lib/polars/utils/various.rb