Contents

require 'spec_helper.rb'

describe Daru::DataFrame do
  context ".from_csv" do
    it "loads from a CSV file" do
      df = Daru::DataFrame.from_csv('spec/fixtures/matrix_test.csv', 
        col_sep: ' ', headers: true)

      expect(df.vectors).to eq([:image_resolution, :mls, :true_transform].to_index)
      expect(df.vector[:image_resolution].first).to eq(6.55779)
      expect(df.vector[:true_transform].first).to eq("-0.2362347,0.6308649,0.7390552,0,0.6523478,-0.4607318,0.6018043,0,0.7201635,0.6242881,-0.3027024,4262.65,0,0,0,1")
    end
  end

  context "JSON" do
    it "loads parsed JSON" do
      require 'json'

      json = File.read 'spec/fixtures/countries.json'
      df   = Daru::DataFrame.new JSON.parse(json)

      expect(df.vectors).to eq([
        :name, :nativeName, :tld, :cca2, :ccn3, :cca3, :currency, :callingCode, 
        :capital, :altSpellings, :relevance, :region, :subregion, :language, 
        :languageCodes, :translations, :latlng, :demonym, :borders, :area].to_index)

      expect(df.row[0][:name]).to eq("Afghanistan")
    end
  end

  context "#inspect" do
    it "prints DataFrame pretty" do

    end
  end

  context "#to_csv" do
    # TODO
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
daru-0.0.5	spec/io/io_spec.rb