stages 0.2.3 → 0.2.4

data/VERSION

@@ -1 +1 @@
-0.2.3
+0.2.4

data/examples/sing_custom_stages.rb

@@ -38,7 +38,7 @@ def setup_pipeline
   
   each_character = Each.new{ |x| x.chars }
   trim_whitespace = Select.new{ |x| x != ' '}
-  letters_in_each_line = Wrap.new(get_lyric | each_character | trim_whitespace)
+  letters_in_each_line = Wrap.new(get_lyric | each_character | trim_whitespace, :array)
   each_letter = Each.new
   each_note = Generator.new(@lyrics.keys)
   count_everything = Count.new

data/examples/sing_subpipes.rb

@@ -31,7 +31,7 @@ puts setup_pipeline.run.inspect
 
 puts "one line at a time"
 lyrics = Each.new(@lyrics.keys)
-letters_in_chunks = Wrap.new(Map.new{ |x| @lyrics[x]} | Each.new{ |x| x.chars}).with_hash
+letters_in_chunks = Wrap.new(Map.new{ |x| @lyrics[x]} | Each.new{ |x| x.chars})
 each_letter = Each.new{ |x| x.values.first }
 count = Count.new
 

data/lib/stages/wrap.rb

@@ -1,27 +1,19 @@
 module Stages
   class Wrap < Stage    
-    def initialize(pipeline)
+    def initialize(pipeline, *args)
       @pipeline = pipeline
       @output_style = :hash
-      @aggregated = false
+      unless args.empty?
+        if args.include? :array
+          @output_style = :array        
+        elsif args.include? :each
+          @output_style = :each
+        end
+        @aggregated = true if args.include? :aggregated
+      end
       super()
     end
-    
-    def array
-      @output_style = :array
-      self
-    end
-    
-    def each
-      @output_style = :each
-      self
-    end
-    
-    def aggregated
-      @aggregated = true
-      self
-    end
-    
+      
     def process
       while value = input
         subpipe = Emit.new(value) | @pipeline

data/stages.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = "stages"
-  s.version = "0.2.3"
+  s.version = "0.2.4"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["The Justice Eight"]

data/test/test_stages.rb

@@ -78,7 +78,7 @@ class TestStages < MiniTest::Unit::TestCase
   end
   
   test 'hash/aggregated wrap mode wrap' do
-    pipeline = Each.new(%w(foo bar)) | Wrap.new(Each.new{ |x| x.chars} | Count.new).aggregated
+    pipeline = Each.new(%w(foo bar)) | Wrap.new(Each.new{ |x| x.chars} | Count.new, :aggregated)
     result = pipeline.run
     assert_equal(2, result['foo']['o'])
     result = pipeline.run
@@ -87,7 +87,7 @@ class TestStages < MiniTest::Unit::TestCase
   
   
   test 'array mode wrap' do
-    pipeline = Each.new(%w(foo bar)) | Wrap.new(Each.new{ |x| x.chars}).array
+    pipeline = Each.new(%w(foo bar)) | Wrap.new(Each.new{ |x| x.chars}, :array)
     result = pipeline.run
     assert_equal(%w(f o o), result)
     result = pipeline.run
@@ -95,7 +95,7 @@ class TestStages < MiniTest::Unit::TestCase
   end
   
   test 'each mode wrap' do
-    pipeline = Each.new(%w(foo bar)) | Wrap.new(Each.new{ |x| x.chars}).each
+    pipeline = Each.new(%w(foo bar)) | Wrap.new(Each.new{ |x| x.chars}, :each)
     expected = %w(r a b o o f)
     while r = pipeline.run
       assert_equal(expected.pop, r)

metadata

@@ -2,7 +2,7 @@
 name: stages
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.2.3
+  version: 0.2.4
 platform: ruby
 authors: 
 - The Justice Eight
@@ -70,7 +70,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements: 
   - - ">="
     - !ruby/object:Gem::Version 
-      hash: 1416732163094176544
+      hash: 940714069352004572
       segments: 
       - 0
       version: "0"