rdatasets 0.3.2 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d83d00803d71e447c6d4ec666cc256619d406341aabdecf8e32d141c65ee42b7
-  data.tar.gz: 4829511ff20f1dc8482f517ee0e49cadbe7c70d85b1f08198561dd6aee630492
+  metadata.gz: 4885051665cb07837095ed6c509b1fb6b701fc5646060c4002fd21e21bd82ad5
+  data.tar.gz: ea511278964d6fe0f460cce77f226d1f6dcab72e8ffde816e4651d3513e66498
 SHA512:
-  metadata.gz: eea59ee442f3d3a26188942fa8b7689600152dfe8fda583720afc3dfd139f3d9d306ae4c246d3c685ed01d7c3b9d39f025a8ad2a802da5bce290e5e804ef7904
-  data.tar.gz: 46ff298cbce5e13b90b9fbb0c97346de5368a0b8abc819e0341642e06296e109ebd941309f5c46851d5b04fa113cf17436ad523199eb6fe7c3d607697fc5ce85
+  metadata.gz: 0507f82fbf2eb9cd86b59489ddf3ce8bd63163bbcf13b0979c78f8f3c5f3c4776a2c7a2ae52b2d7106aa92f736092a1435db5b28f5d70037e60ebe2652c68509
+  data.tar.gz: fdea2c475cede1d2718978611cd84b5966659351a806a0d83047c83b6e1b7fe9fbc269b4fe752f5936271d69092a0509e071fb2712befdfc3657f25e4879a422

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    rdatasets (0.3.2)
+    rdatasets (0.4.0)
       daru
 
 GEM
@@ -14,6 +14,9 @@ GEM
     diff-lcs (1.3)
     packable (1.3.10)
       backports
+    parallel (1.13.0)
+    parallel_tests (2.27.1)
+      parallel
     rake (10.5.0)
     rspec (3.8.0)
       rspec-core (~> 3.8.0)
@@ -35,6 +38,7 @@ PLATFORMS
 
 DEPENDENCIES
   bundler (~> 2.0)
+  parallel_tests
   rake (~> 10.0)
   rdatasets!
   rspec (~> 3.0)

data/README.md

@@ -8,6 +8,8 @@ This ruby gem allows you to access over 1200 datasets included in R from Ruby.
 - All the datasets were ported from [RDatasets](https://github.com/vincentarelbundock/Rdatasets) created by Vincent.
 - This Ruby gem was inspired by [RDatasets.jl](https://github.com/johnmyleswhite/RDatasets.jl) created by John Myles White.
 
+under development
+
 ## Installation
 
 ```bash

data/lib/rdatasets.rb

@@ -9,9 +9,15 @@ module RDatasets
   # @param package_name [String, Symbol] :R package name
   # @param dataset_name [String, Symbol] :R dataset name
   # @return [Daru::DataFrame]
-  def load(package_name, dataset_name)
-    file_path = filepath(package_name, dataset_name)
-    Daru::DataFrame.from_csv(file_path)
+  def load(package_name, dataset_name = nil)
+    if dataset_name
+      file_path = filepath(package_name, dataset_name)
+      df = Daru::DataFrame.from_csv(file_path)
+      df.set_index(df[0].name) if original_index_is_sequential? df
+      df
+    else
+      packages(package_name)
+    end
   end
 
   # Get the file path of a certain dataset.
@@ -26,14 +32,40 @@ module RDatasets
     # "car" package directory is a symbolic link.
     # Do not use Symbolic links because they can cause error on Windows.
     package_name = 'carData' if package_name == 'car'
-    dataset_name << '.csv'
+    dataset_name += '.csv'
     File.join(rdata_directory, package_name, dataset_name)
   end
 
-  def packages
+  # Display information of all data sets.
+  # @return [Daru::DataFrame]
+  def datasets
     file_path = File.expand_path('../data/datasets.csv', __dir__)
     Daru::DataFrame.from_csv(file_path)
   end
+
+  #
+  # @overload packages
+  #   Show a list of all packages.
+  #   @return [Array<Symbol>]
+  # @overload packages(package_name)
+  #   Show a list of datasets included in the package.
+  #   @param [String, Symbol] :R package name
+  #   @return [Array<Symbol>]
+  def packages(package_name = nil)
+    if package_name
+      df = datasets
+      ds = df.where(df['Package'].eq package_name.to_s)
+      ds['Item'].to_a.map(&:to_sym)
+    else
+      datasets['Package'].to_a.uniq.map(&:to_sym)
+    end
+  end
+
+  # Check if the index of original r dataset is sequential.
+  def original_index_is_sequential?(df)
+    df[0].to_a == [1..df.size]
+  end
+  private_class_method :original_index_is_sequential?
 end
 
 module Daru

data/lib/rdatasets/version.rb

@@ -1,3 +1,3 @@
 module RDatasets
-  VERSION = '0.3.2'.freeze
+  VERSION = '0.4.0'.freeze
 end

data/rdatasets.gemspec

@@ -25,6 +25,7 @@ Gem::Specification.new do |spec|
   spec.add_dependency 'daru'
 
   spec.add_development_dependency 'bundler', '~> 2.0'
+  spec.add_development_dependency 'parallel_tests'
   spec.add_development_dependency 'rake', '~> 10.0'
   spec.add_development_dependency 'rspec', '~> 3.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rdatasets
 version: !ruby/object:Gem::Version
-  version: 0.3.2
+  version: 0.4.0
 platform: ruby
 authors:
 - kojix2
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-01-17 00:00:00.000000000 Z
+date: 2019-01-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: daru
@@ -38,6 +38,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: parallel_tests
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
@@ -76,9 +90,9 @@ files:
 - ".gitignore"
 - ".rspec"
 - ".travis.yml"
-- GPL-3
 - Gemfile
 - Gemfile.lock
+- LICENSE.md
 - README.md
 - Rakefile
 - data/COUNT/affairs.csv
@@ -1350,8 +1364,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.7.6
+rubygems_version: 3.0.1
 signing_key: 
 specification_version: 4
 summary: Ruby package for loading classical data sets vailable in R