daru_lite 0.1.3 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fe2a3da2352dd68322a62e048c7870ad34f243a31bcd0cf18725b15583e4c5e8
-  data.tar.gz: 889d0714e1188240f43dc741f2b58b0ce3fe8731a07b39ee4d366f54ce301f6b
+  metadata.gz: 6da8c79960f009b08e12c9d252a267e4d69d6c2152ded63fbe8180891eb25b86
+  data.tar.gz: 074d65df9bbc034b37089cf0938a5df45006f8e264ed259721d5823e0ae41c3b
 SHA512:
-  metadata.gz: 370f64692790e1661642c2e40ec4b56baa8fd8ac000be5623293013b169b7d96c0e5d5fe122801bbea40945761710b0b94c148fae27c67a42ab1825683416876
-  data.tar.gz: ebb89b2307548a0deaeda39e8e72a899a0f4e20c0e5a5ee16515c7aaea5ac95e8f7da3dc752b80a7dba64ec2f4e19717ba3703c06ee0793111c36da968ed5b02
+  metadata.gz: 56055dcbc442963a8c9881d0b86c767cf9d49d382b1f78b33e7603d83a496028116fef819b38fd15110f67bacccf143354efdba9a919fe829d54006c2e0e6bd5
+  data.tar.gz: b3b20cfa80cd6ff336979d1ec836673de99f828883d45186a2762b396ab85ed7b57c8b75709b9ff12623121ed34c707ad66b3ad26aa530e98f43aa5fb27dd4a0

data/.github/dependabot.yml

@@ -0,0 +1,11 @@
+# To get started with Dependabot version updates, you'll need to specify which
+# package ecosystems to update and where the package manifests are located.
+# Please see the documentation for all configuration options:
+# https://docs.github.com/code-security/dependabot/dependabot-version-updates/configuration-options-for-the-dependabot.yml-file
+
+version: 2
+updates:
+  - package-ecosystem: "bundler" # See documentation for possible values
+    directory: "/" # Location of package manifests
+    schedule:
+      interval: "daily"

data/.github/workflows/ci.yml

@@ -1,15 +1,6 @@
 name: CI
 on: [push]
 
-env:
-  CC_TEST_REPORTER_ID: ${{secrets.CC_TEST_REPORTER_ID}}
-  # `github.ref` points to the *merge commit* when running tests on a pull request, which will be a commit
-  # that doesn't exists in our code base. Since this workflow triggers from a PR, we use the HEAD SHA instead.
-  #
-  # NOTE: These are both used by Code Climate (cc-test-reporter).
-  GIT_COMMIT_SHA: ${{github.event.pull_request.head.sha}}
-  GIT_BRANCH: ${{github.head_ref}}
-
 jobs:
   lint:
     runs-on: ubuntu-latest
@@ -19,7 +10,7 @@ jobs:
     - name: Set up Ruby
       uses: ruby/setup-ruby@v1
       with:
-        ruby-version: 3.1
+        ruby-version: 3.4
         bundler-cache: true # runs 'bundle install' and caches installed gems automatically
     - name: Run linters
       run: |
@@ -29,7 +20,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        ruby-version: ['3.1']
+        ruby-version: ['3.4']
 
     steps:
     - uses: actions/checkout@v3
@@ -38,16 +29,5 @@ jobs:
       with:
         ruby-version: ${{ matrix.ruby-version }}
         bundler-cache: true # runs 'bundle install' and caches installed gems automatically
-    - name: "Download cc-test-reporter from codeclimate.com"
-      run: |
-        curl -L https://codeclimate.com/downloads/test-reporter/test-reporter-latest-linux-amd64 > ./cc-test-reporter
-        chmod +x ./cc-test-reporter
-    - name: "Report to Code Climate that we will send a coverage report."
-      run: ./cc-test-reporter before-build
     - name: Run tests
       run: bundle exec rspec
-    - name: Upload code coverage to Code Climate
-      run: |
-        ./cc-test-reporter after-build \
-          --coverage-input-type simplecov \
-          ./coverage/.resultset.json

data/.rubocop.yml

@@ -16,10 +16,10 @@ AllCops:
     - 'profile/*'
     - 'tmp/*'
   DisplayCopNames: true
-  TargetRubyVersion: 2.7
+  TargetRubyVersion: 3.4
   NewCops: enable
 
-require:
+plugins:
   - rubocop-performance
   - rubocop-rspec
 

data/.rubocop_todo.yml

@@ -80,7 +80,7 @@ Naming/MethodParameterName:
 # ForbiddenPrefixes: is_, has_, have_
 # AllowedMethods: is_a?
 # MethodDefinitionMacros: define_method, define_singleton_method
-Naming/PredicateName:
+Naming/PredicatePrefix:
   Exclude:
     - 'spec/**/*'
     - 'lib/daru_lite/data_frame/missable.rb'

data/README.md

@@ -4,8 +4,6 @@ Simple, straightforward DataFrames for Ruby
 
 [![Build Status](https://github.com/pollandroll/daru_lite/actions/workflows/build.yml/badge.svg)](https://github.com/pollandroll/daru_lite/actions)
 [![Gem Version](https://img.shields.io/gem/v/daru_lite.svg)](https://rubygems.org/gems/daru_lite)
-[![Maintainability](https://api.codeclimate.com/v1/badges/f87d4ed10b5731e50184/maintainability)](https://codeclimate.com/github/pollandroll/daru_lite/maintainability)
-[![Test Coverage](https://api.codeclimate.com/v1/badges/f87d4ed10b5731e50184/test_coverage)](https://codeclimate.com/github/pollandroll/daru_lite/test_coverage)
 
 ## Introduction
 

data/benchmarks/db_loading.rb

@@ -3,7 +3,6 @@ $:.unshift File.expand_path("../../lib", __FILE__)
 require 'benchmark'
 require 'daru_lite'
 require 'sqlite3'
-require 'dbi'
 require 'active_record'
 
 db_name = 'daru_lite_test.sqlite'

data/daru_lite.gemspec

@@ -28,29 +28,27 @@ Gem::Specification.new do |spec|
   spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
   spec.require_paths = ["lib"]
 
-  spec.add_development_dependency 'activerecord', '~> 6.0'
+  spec.add_development_dependency 'activerecord', '~> 8.0'
   spec.add_development_dependency 'awesome_print', '~> 1.9.2'
-  spec.add_development_dependency 'bundler', '~> 2.1', '>= 2.1.4'
-  spec.add_development_dependency 'dbd-sqlite3', '~> 1.2.5'
-  spec.add_development_dependency 'dbi', '~> 0.4.5'
+  spec.add_development_dependency 'bundler', '~> 2.6'
+  spec.add_development_dependency 'csv', '~> 3.3.4'
+  spec.add_development_dependency 'debug', '>= 1.0.0'
   spec.add_development_dependency 'distribution', '~> 0.8'
   spec.add_development_dependency 'matrix', '~> 0.4.2'
-  spec.add_development_dependency 'nokogiri', '~> 1.16.2'
+  spec.add_development_dependency 'nokogiri', '~> 1.18.0'
   spec.add_development_dependency 'prime', '~> 0.1.2'
-  spec.add_development_dependency 'pry', '~> 0.14'
-  spec.add_development_dependency 'pry-byebug', '~> 3.10.1'
   spec.add_development_dependency 'rake', '~> 13.0'
   spec.add_development_dependency 'rspec', '~> 3.11'
-  spec.add_development_dependency 'rspec-its', '~> 1.3.0'
+  spec.add_development_dependency 'rspec-its', '~> 2.0.0'
   spec.add_development_dependency 'rubocop', '~> 1.60'
-  spec.add_development_dependency 'rubocop-performance', '~> 1.20.2'
-  spec.add_development_dependency 'rubocop-rspec', '~> 2.25'
+  spec.add_development_dependency 'rubocop-performance', '~> 1.25.0'
+  spec.add_development_dependency 'rubocop-rspec', '~> 3.6'
   spec.add_development_dependency 'ruby-prof', '~> 1.7.0'
   spec.add_development_dependency 'simplecov', '~> 0.22.0'
   spec.add_development_dependency 'simplecov_json_formatter', '~> 0.1.4'
   spec.add_development_dependency 'spreadsheet', '~> 1.3.0'
-  spec.add_development_dependency 'sqlite3', '~> 1.7.2'
+  spec.add_development_dependency 'sqlite3', '~> 2.6.0'
   # issue : https://github.com/SciRuby/daru/issues/493 occured
   # with latest version of sqlite3
-  spec.add_development_dependency 'webmock', '~> 3.20.0'
+  spec.add_development_dependency 'webmock', '~> 3.25.0'
 end

data/lib/daru_lite/accessors/array_wrapper.rb

@@ -7,13 +7,13 @@ module DaruLite
 
       def_delegators :@data, :slice!
 
-      def each(&block)
-        @data.each(&block)
+      def each(&)
+        @data.each(&)
         self
       end
 
-      def map!(&block)
-        @data.map!(&block)
+      def map!(&)
+        @data.map!(&)
         self
       end
 

data/lib/daru_lite/core/merge.rb

@@ -1,6 +1,14 @@
 module DaruLite
   module Core
     class MergeFrame
+      LEFT_RIGHT_COMBINATIONS = {
+        #       left   right
+        inner: [false, false],
+        left: [true, false],
+        right: [false, true],
+        outer: [true, true]
+      }.freeze
+
       class NilSorter
         include Comparable
 
@@ -55,14 +63,6 @@ module DaruLite
 
       attr_accessor :merge_key
 
-      LEFT_RIGHT_COMBINATIONS = {
-        #       left   right
-        inner: [false, false],
-        left: [true, false],
-        right: [false, true],
-        outer: [true, true]
-      }.freeze
-
       def init_opts(opts)
         @on = opts[:on]
         @keep_left, @keep_right = extract_left_right(opts[:how])

data/lib/daru_lite/data_frame/calculatable.rb

@@ -110,8 +110,8 @@ module DaruLite
       #   # 4 555
       #   # 5 666
       #   # 6 777
-      def vector_by_calculation(&block)
-        a = each_row.map { |r| r.instance_eval(&block) }
+      def vector_by_calculation(&)
+        a = each_row.map { |r| r.instance_eval(&) }
 
         DaruLite::Vector.new a, index: @index
       end

data/lib/daru_lite/data_frame/filterable.rb

@@ -68,8 +68,8 @@ module DaruLite
       #   df.filter(:row) do |row|
       #     row[:a] + row[:d] < 100
       #   end
-      def filter(axis = :vector, &block)
-        dispatch_to_axis_pl axis, :filter, &block
+      def filter(axis = :vector, &)
+        dispatch_to_axis_pl(axis, :filter, &)
       end
 
       # Returns a dataframe in which rows with any of the mentioned values
@@ -113,8 +113,8 @@ module DaruLite
       end
 
       # creates a new vector with the data of a given field which the block returns true
-      def filter_vector(vec, &block)
-        DaruLite::Vector.new(each_row.select(&block).map { |row| row[vec] })
+      def filter_vector(vec, &)
+        DaruLite::Vector.new(each_row.select(&).map { |row| row[vec] })
       end
 
       # Iterates over each row and retains it in a new DataFrame if the block returns

data/lib/daru_lite/data_frame/i_o_able.rb

@@ -32,8 +32,8 @@ module DaruLite
         #     :col_sep           => ',',
         #     :converters        => :numeric
         #   }
-        def from_csv(path, opts = {}, &block)
-          DaruLite::IO.from_csv path, opts, &block
+        def from_csv(path, opts = {}, &)
+          DaruLite::IO.from_csv(path, opts, &)
         end
 
         # Read data from an Excel file into a DataFrame.
@@ -45,28 +45,30 @@ module DaruLite
         # == Options
         #
         # *:worksheet_id - ID of the worksheet that is to be read.
-        def from_excel(path, opts = {}, &block)
-          DaruLite::IO.from_excel path, opts, &block
+        def from_excel(path, opts = {}, &)
+          DaruLite::IO.from_excel(path, opts, &)
         end
 
         # Read a database query and returns a Dataset
         #
-        # @param dbh [DBI::DatabaseHandle, String] A DBI connection OR Path to a SQlite3 database.
+        # @param arh [ActiveRecord::ConnectionAdapters::AbstractAdapter, String] An ActiveRecord connection
+        # OR Path to a SQlite3 database.
         # @param query [String] The query to be executed
         #
         # @return A dataframe containing the data resulting from the query
         #
         # USE:
         #
-        #  dbh = DBI.connect("DBI:Mysql:database:localhost", "user", "password")
+        #  ActiveRecord::Base.establish_connection(adapter: 'sqlite3', database: "path/to/sqlite.db")
+        #  arh = ActiveRecord::Base.connection
         #  DaruLite::DataFrame.from_sql(dbh, "SELECT * FROM test")
         #
         #  #Alternatively
         #
-        #  require 'dbi'
+        #  require 'active_record'
         #  DaruLite::DataFrame.from_sql("path/to/sqlite.db", "SELECT * FROM test")
-        def from_sql(dbh, query)
-          DaruLite::IO.from_sql dbh, query
+        def from_sql(arh, query)
+          DaruLite::IO.from_sql arh, query
         end
 
         # Read a dataframe from AR::Relation
@@ -149,16 +151,17 @@ module DaruLite
       #
       # == Arguments
       #
-      # * dbh - DBI database connection object.
+      # * arh - ActiveRecord database connection object.
       # * query - Query string.
       #
       # == Usage
       #
       #  ds = DaruLite::DataFrame.new({:id=>DaruLite::Vector.new([1,2,3]), :name=>DaruLite::Vector.new(["a","b","c"])})
-      #  dbh = DBI.connect("DBI:Mysql:database:localhost", "user", "password")
-      #  ds.write_sql(dbh,"test")
-      def write_sql(dbh, table)
-        DaruLite::IO.dataframe_write_sql self, dbh, table
+      #  ActiveRecord::Base.establish_connection(adapter: 'sqlite3', database: "path/to/sqlite.db")
+      #  arh = ActiveRecord::Base.connection
+      #  ds.write_sql(arh,"test")
+      def write_sql(arh, table)
+        DaruLite::IO.dataframe_write_sql self, arh, table
       end
 
       # Use marshalling to save dataframe to a file.

data/lib/daru_lite/data_frame/iterable.rb

@@ -68,8 +68,8 @@ module DaruLite
       #
       # * +axis+ - The axis to iterate over. Can be :vector (or :column)
       # or :row. Default to :vector.
-      def each(axis = :vector, &block)
-        dispatch_to_axis axis, :each, &block
+      def each(axis = :vector, &)
+        dispatch_to_axis(axis, :each, &)
       end
 
       # Iterate over a row or vector and return results in a DaruLite::Vector.
@@ -87,8 +87,8 @@ module DaruLite
       #
       # * +axis+ - The axis to iterate over. Can be :vector (or :column)
       # or :row. Default to :vector.
-      def collect(axis = :vector, &block)
-        dispatch_to_axis_pl axis, :collect, &block
+      def collect(axis = :vector, &)
+        dispatch_to_axis_pl(axis, :collect, &)
       end
 
       # Map over each vector or row of the data frame according to
@@ -107,8 +107,8 @@ module DaruLite
       #
       # * +axis+ - The axis to map over. Can be :vector (or :column) or :row.
       # Default to :vector.
-      def map(axis = :vector, &block)
-        dispatch_to_axis_pl axis, :map, &block
+      def map(axis = :vector, &)
+        dispatch_to_axis_pl(axis, :map, &)
       end
 
       # Destructive map. Modifies the DataFrame. Each run of the block
@@ -119,11 +119,11 @@ module DaruLite
       #
       # * +axis+ - The axis to map over. Can be :vector (or :column) or :row.
       # Default to :vector.
-      def map!(axis = :vector, &block)
+      def map!(axis = :vector, &)
         if %i[vector column].include?(axis)
-          map_vectors!(&block)
+          map_vectors!(&)
         elsif axis == :row
-          map_rows!(&block)
+          map_rows!(&)
         end
       end
 
@@ -144,8 +144,8 @@ module DaruLite
       #
       # * +axis+ - The axis to map over. Can be :vector (or :column) or :row.
       # Default to :vector.
-      def recode(axis = :vector, &block)
-        dispatch_to_axis_pl axis, :recode, &block
+      def recode(axis = :vector, &)
+        dispatch_to_axis_pl(axis, :recode, &)
       end
 
       # Replace specified values with given value

data/lib/daru_lite/data_frame/joinable.rb

@@ -22,7 +22,10 @@ module DaruLite
         df = row[*(@index.to_a - other_df.index.to_a)]
 
         df = df.concat(other_df)
-        df.index = DaruLite::Index.new(index)
+        df.index = @index.class.public_send(
+          @index.is_a?(DaruLite::MultiIndex) ? :from_tuples : :new,
+          index
+        )
         df
       end
 

data/lib/daru_lite/data_frame/queryable.rb

@@ -31,9 +31,9 @@ module DaruLite
       #   df.any?(:row) do |row|
       #     row[:a] < 3 and row[:b] == 'b'
       #   end #=> true
-      def any?(axis = :vector, &block)
+      def any?(axis = :vector, &)
         if %i[vector column].include?(axis)
-          @data.any?(&block)
+          @data.any?(&)
         elsif axis == :row
           each_row do |row|
             return true if yield(row)
@@ -53,11 +53,11 @@ module DaruLite
       #   df.all?(:row) do |row|
       #     row[:a] < 10
       #   end #=> true
-      def all?(axis = :vector, &block)
+      def all?(axis = :vector, &)
         if %i[vector column].include?(axis)
-          @data.all?(&block)
+          @data.all?(&)
         elsif axis == :row
-          each_row.all?(&block)
+          each_row.all?(&)
         else
           raise ArgumentError, "Unidentified axis #{axis}"
         end

data/lib/daru_lite/dataframe.rb

@@ -38,6 +38,8 @@ module DaruLite
     include DaruLite::Maths::Arithmetic::DataFrame
     include DaruLite::Maths::Statistics::DataFrame
 
+    AXES = %i[row vector].freeze
+
     attr_accessor(*Configuration::INSPECT_OPTIONS_KEYS)
 
     extend Gem::Deprecate
@@ -478,15 +480,17 @@ module DaruLite
       self
     end
 
-    def method_missing(name, *args, &block)
-      if /(.+)=/.match?(name)
-        name = name[/(.+)=/].delete('=')
+    def method_missing(name, *args, &)
+      stringified_name = name.to_s
+
+      if /^([^=]+)=/.match?(stringified_name)
+        name = stringified_name[/^([^=]+)=/].delete('=')
         name = name.to_sym unless has_vector?(name)
         insert_or_modify_vector [name], args[0]
       elsif has_vector?(name)
         self[name]
-      elsif has_vector?(name.to_s)
-        self[name.to_s]
+      elsif has_vector?(stringified_name)
+        self[stringified_name]
       else
         super
       end
@@ -527,28 +531,26 @@ module DaruLite
       end
     end
 
-    def dispatch_to_axis(axis, method, *args, &block)
+    def dispatch_to_axis(axis, method, *, &)
       if %i[vector column].include?(axis)
-        send(:"#{method}_vector", *args, &block)
+        send(:"#{method}_vector", *, &)
       elsif axis == :row
-        send(:"#{method}_row", *args, &block)
+        send(:"#{method}_row", *, &)
       else
         raise ArgumentError, "Unknown axis #{axis}"
       end
     end
 
-    def dispatch_to_axis_pl(axis, method, *args, &block)
+    def dispatch_to_axis_pl(axis, method, *, &)
       if %i[vector column].include?(axis)
-        send(:"#{method}_vectors", *args, &block)
+        send(:"#{method}_vectors", *, &)
       elsif axis == :row
-        send(:"#{method}_rows", *args, &block)
+        send(:"#{method}_rows", *, &)
       else
         raise ArgumentError, "Unknown axis #{axis}"
       end
     end
 
-    AXES = %i[row vector].freeze
-
     def extract_axis(names, default = :vector)
       if AXES.include?(names.last)
         names.pop

data/lib/daru_lite/date_time/index.rb

@@ -30,7 +30,7 @@ module DaruLite
         (
           (?<offset>MONTH|YEAR|S|H|MB|ME|M|D|YB|YE) |
           (?<offset>W)(-(?<weekday>#{DOW_REGEXP}))?
-        )$/x.freeze
+        )$/x
 
       # Generates a DaruLite::DateOffset object for generic offsets or one of the
       # specialized classed within DaruLite::Offsets depending on the 'frequency'
@@ -131,7 +131,7 @@ module DaruLite
         end
       end
 
-      DATE_PRECISION_REGEXP = /^(\d\d\d\d)(-\d{1,2}(-\d{1,2}( \d{1,2}(:\d{1,2}(:\d{1,2})?)?)?)?)?$/.freeze
+      DATE_PRECISION_REGEXP = /^(\d\d\d\d)(-\d{1,2}(-\d{1,2}( \d{1,2}(:\d{1,2}(:\d{1,2})?)?)?)?)?$/
       DATE_PRECISIONS = [nil, :year, :month, :day, :hour, :min, :sec].freeze
 
       def determine_date_precision_of(date_string)
@@ -222,8 +222,8 @@ module DaruLite
       new(source, freq: :infer) if source && ArrayHelper.array_of?(source, ::DateTime)
     end
 
-    def each(&block)
-      to_a.each(&block)
+    def each(&)
+      to_a.each(&)
     end
 
     attr_reader :frequency, :offset, :periods, :keys
@@ -367,20 +367,20 @@ module DaruLite
       end
     end
 
-    def pos(*args)
+    def pos(*)
       # to filled
-      out = self[*args]
+      out = self[*]
       return out if out.is_a? Numeric
 
       out.map { |date| self[date] }
     end
 
-    def subset(*args)
-      self[*args]
+    def subset(*)
+      self[*]
     end
 
-    def valid?(*args)
-      self[*args]
+    def valid?(*)
+      self[*]
       true
     rescue IndexError
       false

data/lib/daru_lite/extensions/which_dsl.rb

@@ -12,8 +12,8 @@ module DaruLite
     # df.which{ (`NameTitle` == 'Dr') & (`Sex` == 'female') }
     # equals
     # df.where( df['NameTitle'].eq('Dr') & df['Sex'].eq('female') )
-    def which(&block)
-      WhichQuery.new(self, &block).exec
+    def which(&)
+      WhichQuery.new(self, &).exec
     end
   end
 

data/lib/daru_lite/index/index.rb

@@ -20,7 +20,7 @@ module DaruLite
 
     # We over-ride the .new method so that any sort of Index can be generated
     # from DaruLite::Index based on the types of arguments supplied.
-    def self.new(*args, &block)
+    def self.new(*args, &)
       # FIXME: I'm not sure this clever trick really deserves our attention.
       # Most of common ruby libraries just avoid it in favor of usual
       # factor method, like `Index.create`. When `Index.new(...).class != Index`
@@ -29,7 +29,7 @@ module DaruLite
 
       MultiIndex.try_from_tuples(source) ||
         DateTimeIndex.try_create(source) ||
-        allocate.tap { |i| i.send :initialize, *args, &block }
+        allocate.tap { |i| i.send(:initialize, *args, &) }
     end
 
     def self.coerce(maybe_index)
@@ -229,7 +229,7 @@ module DaruLite
     #   #     2  false
     #   #     3  false
     #   #     4  true
-    def is_values(*indexes) # rubocop:disable Naming/PredicateName
+    def is_values(*indexes) # rubocop:disable Naming/PredicatePrefix
       bool_array = @keys.map { |r| indexes.include?(r) }
       DaruLite::Vector.new(bool_array)
     end
@@ -377,7 +377,7 @@ module DaruLite
     def numeric_pos(key)
       if @relation_hash.key?(key)
         @relation_hash[key]
-      elsif key.is_a?(Numeric) && (key < size && key >= -size)
+      elsif key.is_a?(Numeric) && key < size && key >= -size
         key
       else
         raise IndexError, "Specified index #{key.inspect} does not exist"

data/lib/daru_lite/index/multi_index.rb

@@ -1,11 +1,11 @@
 module DaruLite
   class MultiIndex < Index # rubocop:disable Metrics/ClassLength
-    def each(&block)
-      to_a.each(&block)
+    def each(&)
+      to_a.each(&)
     end
 
-    def map(&block)
-      to_a.map(&block)
+    def map(&)
+      to_a.map(&)
     end
 
     attr_reader :labels, :name
@@ -122,8 +122,10 @@ module DaruLite
     end
 
     def [](*key)
-      key.flatten!
-      if key[0].is_a?(Range)
+      if key[0].is_a?(Array)
+        collection = key.map { |actual_key| self[*actual_key] }
+        collection.one? ? collection.first : collection
+      elsif key[0].is_a?(Range)
         retrieve_from_range(key[0])
       elsif key[0].is_a?(Integer) && key.size == 1
         try_retrieve_from_integer(key[0])
@@ -158,15 +160,19 @@ module DaruLite
 
         return indexes
       end
-      res = self[indexes]
+      res = self[*indexes]
       return res if res.is_a? Integer
 
       res.map { |i| self[i] }
     end
 
     def subset(*indexes)
-      if indexes.first.is_a? Integer
+      first_index = indexes.first
+      if first_index.is_a? Integer
         MultiIndex.from_tuples(indexes.map { |index| key(index) })
+      elsif first_index.is_a?(Array) && include?(first_index)
+        # Same logic as in DaruLite::Index#subset
+        MultiIndex.from_tuples indexes
       else
         self[indexes].conform indexes
       end