JezK
Edit File: tc_data_converters.rb
#!/usr/local/bin/ruby -w # tc_data_converters.rb # # Created by James Edward Gray II on 2005-12-30. # Copyright 2005 Gray Productions. All rights reserved. require "test/unit" require "faster_csv" class TestDataConverters < Test::Unit::TestCase def setup @data = "Numbers,:integer,1,:float,3.015" @parser = FasterCSV.new(@data) @custom = lambda { |field| field =~ /\A:(\S.*?)\s*\Z/ ? $1.to_sym : field } @win_safe_time_str = Time.now.strftime("%a %b %d %H:%M:%S %Y") end def test_builtin_integer_converter # does convert [-5, 1, 10000000000].each do |n| assert_equal(n, FasterCSV::Converters[:integer][n.to_s]) end # does not convert (%w{junk 1.0} + [""]).each do |str| assert_equal(str, FasterCSV::Converters[:integer][str]) end end def test_builtin_float_converter # does convert [-5.1234, 0, 2.3e-11].each do |n| assert_equal(n, FasterCSV::Converters[:float][n.to_s]) end # does not convert (%w{junk 1..0 .015F} + [""]).each do |str| assert_equal(str, FasterCSV::Converters[:float][str]) end end def test_builtin_date_converter # does convert assert_instance_of( Date, FasterCSV::Converters[:date][@win_safe_time_str.sub(/\d+:\d+:\d+ /, "")] ) # does not convert assert_instance_of(String, FasterCSV::Converters[:date]["junk"]) end def test_builtin_date_time_converter # does convert assert_instance_of( DateTime, FasterCSV::Converters[:date_time][@win_safe_time_str] ) # does not convert assert_instance_of(String, FasterCSV::Converters[:date_time]["junk"]) end def test_convert_with_builtin # setup parser... assert(@parser.respond_to?(:convert)) assert_nothing_raised(Exception) { @parser.convert(:integer) } # and use assert_equal(["Numbers", ":integer", 1, ":float", "3.015"], @parser.shift) setup # reset # setup parser... assert_nothing_raised(Exception) { @parser.convert(:float) } # and use assert_equal(["Numbers", ":integer", 1.0, ":float", 3.015], @parser.shift) end def test_convert_order # floats first, then integers... assert_nothing_raised(Exception) do @parser.convert(:float) @parser.convert(:integer) end # gets us nothing but floats assert_equal( [String, String, Float, String, Float], @parser.shift.map { |field| field.class } ) setup # reset # integers have precendance... assert_nothing_raised(Exception) do @parser.convert(:integer) @parser.convert(:float) end # gives us proper number conversion assert_equal( [String, String, Fixnum, String, Float], @parser.shift.map { |field| field.class } ) end def test_builtin_numeric_combo_converter # setup parser... assert_nothing_raised(Exception) { @parser.convert(:numeric) } # and use assert_equal( [String, String, Fixnum, String, Float], @parser.shift.map { |field| field.class } ) end def test_builtin_all_nested_combo_converter # setup parser... @data << ",#{@win_safe_time_str}" # add a DateTime field @parser = FasterCSV.new(@data) # reset parser assert_nothing_raised(Exception) { @parser.convert(:all) } # and use assert_equal( [String, String, Fixnum, String, Float, DateTime], @parser.shift.map { |field| field.class } ) end def test_convert_with_custom_code # define custom converter... assert_nothing_raised(Exception) do @parser.convert { |field| field =~ /\A:(\S.*?)\s*\Z/ ? $1.to_sym : field } end # and use assert_equal(["Numbers", :integer, "1", :float, "3.015"], @parser.shift) setup # reset # mix built-in and custom... assert_nothing_raised(Exception) { @parser.convert(:numeric) } assert_nothing_raised(Exception) { @parser.convert(&@custom) } # and use assert_equal(["Numbers", :integer, 1, :float, 3.015], @parser.shift) end def test_convert_with_custom_code_using_field_info # define custom converter that uses field information... assert_nothing_raised(Exception) do @parser.convert do |field, info| assert_equal(1, info.line) info.index == 4 ? Float(field).floor : field end end # and use assert_equal(["Numbers", ":integer", "1", ":float", 3], @parser.shift) end def test_convert_with_custom_code_using_field_info_header @parser = FasterCSV.new(@data, :headers => %w{one two three four five}) # define custom converter that uses field header information... assert_nothing_raised(Exception) do @parser.convert do |field, info| info.header == "three" ? Integer(field) * 100 : field end end # and use assert_equal(["Numbers", ":integer", 100, ":float", "3.015"], @parser.shift.fields) end def test_shortcut_interface assert_equal( ["Numbers", ":integer", 1, ":float", 3.015], FasterCSV.parse_line(@data, :converters => :numeric) ) assert_equal( ["Numbers", ":integer", 1, ":float", 3.015], FasterCSV.parse_line( @data, :converters => [ :integer, :float ] ) ) assert_equal( ["Numbers", :integer, 1, :float, 3.015], FasterCSV.parse_line( @data, :converters => [ :numeric, @custom ] ) ) end def test_unconverted_fields [ [ @data, ["Numbers", :integer, 1, :float, 3.015], %w{Numbers :integer 1 :float 3.015} ], ["\n", Array.new, Array.new] ].each do |test, fields, unconverted| row = nil assert_nothing_raised(Exception) do row = FasterCSV.parse_line( test, :converters => [:numeric, @custom], :unconverted_fields => true ) end assert_not_nil(row) assert_equal(fields, row) assert_respond_to(row, :unconverted_fields) assert_equal(unconverted, row.unconverted_fields) end data = <<-END_CSV.gsub(/^\s+/, "") first,second,third 1,2,3 END_CSV row = nil assert_nothing_raised(Exception) do row = FasterCSV.parse_line( data, :converters => :numeric, :unconverted_fields => true, :headers => :first_row ) end assert_not_nil(row) assert_equal([["first", 1], ["second", 2], ["third", 3]], row.to_a) assert_respond_to(row, :unconverted_fields) assert_equal(%w{1 2 3}, row.unconverted_fields) assert_nothing_raised(Exception) do row = FasterCSV.parse_line( data, :converters => :numeric, :unconverted_fields => true, :headers => :first_row, :return_headers => true ) end assert_not_nil(row) assert_equal( [%w{first first}, %w{second second}, %w{third third}], row.to_a ) assert_respond_to(row, :unconverted_fields) assert_equal(%w{first second third}, row.unconverted_fields) assert_nothing_raised(Exception) do row = FasterCSV.parse_line( data, :converters => :numeric, :unconverted_fields => true, :headers => :first_row, :return_headers => true, :header_converters => :symbol ) end assert_not_nil(row) assert_equal( [[:first, "first"], [:second, "second"], [:third, "third"]], row.to_a ) assert_respond_to(row, :unconverted_fields) assert_equal(%w{first second third}, row.unconverted_fields) assert_nothing_raised(Exception) do row = FasterCSV.parse_line( data, :converters => :numeric, :unconverted_fields => true, :headers => %w{my new headers}, :return_headers => true, :header_converters => :symbol ) end assert_not_nil(row) assert_equal( [[:my, "my"], [:new, "new"], [:headers, "headers"]], row.to_a ) assert_respond_to(row, :unconverted_fields) assert_equal(Array.new, row.unconverted_fields) end end