pluckit 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ab364fcad0f735ae899cd58285bedc88d04c4586
-  data.tar.gz: a61477d6eb298051c79f43797b3ab4d556479f1f
+  metadata.gz: 7c06dc75e5f426dc4ad79cfab2c4d1728c29fc7b
+  data.tar.gz: 11c8dfcdb0b4a23aeecc091aa2e1caaa3dde37cd
 SHA512:
-  metadata.gz: ac883f33d3cb65eaccae40d1a1662367b9d785ed1eb87c075288517928c234d2a3e7ad96f8f595fed9fbc23364c5876a668ac616295ca767d1b6016a51161610
-  data.tar.gz: bf29eba3a2feeed14c17af96c39976038da4a4b8ce11794a20b7d5d3395b1b70cf2f4988c56122d76887e50e04c313458469d4dde6191534825a7739ec004a08
+  metadata.gz: 57943971842d5c0b76f0c14716d41040247e3de561e08f17b1634e08c3a648c250b75ec3a197a96a7b9564f06d6b05f2ae39633728ba593567bbb0ae18b382a9
+  data.tar.gz: c1433c1310c53811469c1b301b32a8375bacd150dc9b36debaa66fd0933abeb96a6c47507c4b912edfc55aa5f9db91ae38a090bb834c7d6bf8a708ace72f71e1

data/lib/pluckit.rb

@@ -1,16 +1,10 @@
 module PluckIt
-  VERSION = '0.0.1'
+  VERSION = '0.0.2'
 
   class << self
 
     def pluck v, handle
-      if ([Symbol, String].include? handle.class) and v.respond_to? handle
-        if v.method(handle).arity <= 0
-          v.send handle
-        else
-          v.send handle, v
-        end
-      elsif v.is_a? Array
+      if v.is_a? Array
         if handle.is_a? Regexp
           v.grep handle
         else
@@ -18,6 +12,8 @@ module PluckIt
         end
       elsif v.is_a? Hash
         v[handle]
+      elsif ([Symbol, String].include? handle.class) and v.respond_to? handle
+        v.send handle
       else
         raise ArgumentError.new "invalid handle: #{handle}, for value #{v}"
       end
@@ -32,3 +28,11 @@ class Array
     map {|x| PluckIt.pluck x, handle }
   end
 end
+
+
+class Hash
+  def pluck(handle)
+    res = values.map {|x| PluckIt.pluck x, handle }
+    Hash[keys.zip res]
+  end
+end

data/test/test_array.rb

@@ -16,16 +16,6 @@ class PluckItArrayTest < Minitest::Test
       [ 1, 4, 7 ],
       data.pluck(0)
     )
-
-    assert_equal(
-      [ 1, 4, 7 ],
-      data.pluck(:first)
-    )
-
-    assert_equal(
-      [ 1, 4, 7 ],
-      data.pluck('first')
-    )
   end
 
 
@@ -69,4 +59,21 @@ class PluckItArrayTest < Minitest::Test
   end
 
 
+  class ABC
+    attr_accessor :val
+    def initialize(v) self.val = v end
+  end
+
+  def test_obj
+    assert_equal(
+      [ 1, 2, 3 ],
+      [
+        ABC.new(1),
+        ABC.new(2),
+        ABC.new(3),
+      ].pluck(:val)
+    )
+  end
+
+
 end

data/test/test_hash.rb

@@ -0,0 +1,58 @@
+require 'minitest/autorun'
+require 'pluckit'
+
+
+class PluckItHashTest < Minitest::Test
+
+
+  def test_basic
+    assert_equal(
+      {
+        a: 1,
+        b: 4,
+        c: 7,
+      },
+      {
+        a: [ 1, 2, 3 ],
+        b: [ 4, 5, 6 ],
+        c: [ 7, 8, 9 ],
+      }.pluck(0)
+    )
+
+    assert_equal(
+      {
+        a: 1,
+        b: 2,
+        c: 3,
+      },
+      {
+        a: { x: 1, y: 2 },
+        b: { x: 2, y: 4 },
+        c: { x: 3, y: 6 },
+      }.pluck(:x)
+    )
+  end
+
+
+  class ABC
+    attr_accessor :val
+    def initialize(v) self.val = v end
+  end
+
+  def test_obj
+    assert_equal(
+      {
+        a: 1,
+        b: 2,
+        c: 3,
+      },
+      {
+        a: ABC.new(1),
+        b: ABC.new(2),
+        c: ABC.new(3),
+      }.pluck(:val)
+    )
+  end
+
+
+end

data/test/test_pluck.rb

@@ -0,0 +1,71 @@
+require 'minitest/autorun'
+require 'pluckit'
+
+
+class PluckItTest < Minitest::Test
+
+
+  def test_array
+    data = [ 1, 2, 3 ]
+
+    assert_equal(
+      1,
+      PluckIt.pluck(data, 0)
+    )
+
+    assert_equal(
+      2,
+      PluckIt.pluck(data, 1)
+    )
+
+    assert_equal(
+      [ 1, 2 ],
+      PluckIt.pluck(data, 0..1)
+    )
+
+    assert_equal(
+      [ 'a', 'ba', 'ca' ],
+      PluckIt.pluck(
+        [ 'a', 'ba', 'bb', 'bc', 'ca' ],
+        /a/
+      )
+    )
+  end
+
+
+  def test_hash
+    data = {
+      a: 1,
+      b: 2,
+      c: 3,
+    }
+
+    assert_equal(
+      2,
+      PluckIt.pluck(data, :b)
+    )
+
+    assert_nil(
+      PluckIt.pluck(data, :z)
+    )
+  end
+
+
+  class ABC
+    def foo() 123 end
+  end
+
+  def test_obj
+    assert_equal(
+      123,
+      PluckIt.pluck(ABC.new, :foo)
+    )
+
+    assert_equal(
+      self.class.const_get(:ABC),
+      PluckIt.pluck(ABC.new, :class)
+    )
+  end
+
+
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pluckit
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Daniel Pepper
@@ -46,6 +46,8 @@ extra_rdoc_files: []
 files:
 - lib/pluckit.rb
 - test/test_array.rb
+- test/test_hash.rb
+- test/test_pluck.rb
 homepage: https://github.com/dpep/pluckit
 licenses:
 - MIT
@@ -72,3 +74,5 @@ specification_version: 4
 summary: PluckIt
 test_files:
 - test/test_array.rb
+- test/test_hash.rb
+- test/test_pluck.rb