class StatsampleCodificationTestCase

# File pkg/statsample-1.4.0/test/test_codification.rb, line 4
def initialize(*args)
  v1=%w{run walk,run walking running sleep sleeping,dreaming sleep,dream}.to_vector
  @dict={'run'=>'r','walk'=>'w','walking'=>'w','running'=>'r','sleep'=>'s', 'sleeping'=>'s', 'dream'=>'d', 'dreaming'=>'d'}
  @ds={"v1"=>v1}.to_dataset
  super
end

# File pkg/statsample-1.4.0/test/test_codification.rb, line 17
def test_create_excel
  filename=Dir::tmpdir+"/test_excel"+Time.now().to_s+".xls"
  #filename = Tempfile.new("test_codification_"+Time.now().to_s)
  Statsample::Codification.create_excel(@ds, ['v1'], filename)
  field=(["v1"]*8).to_vector
  keys=%w{dream dreaming run running sleep sleeping walk walking}.to_vector
  ds=Statsample::Excel.read(filename)
  assert_equal(field, ds['field'])
  assert_equal(keys, ds['original'])
  assert_equal(keys, ds['recoded'])
  hash=Statsample::Codification.excel_to_recoded_hash(filename)
  assert_equal(keys.data, hash['v1'].keys.sort)
  assert_equal(keys.data, hash['v1'].values.sort)

end

# File pkg/statsample-1.4.0/test/test_codification.rb, line 10
def test_create_hash
  expected_keys_v1=%w{run walk walking running sleep sleeping dream dreaming}.sort
  hash=Statsample::Codification.create_hash(@ds,['v1'])
  assert_equal(['v1'],hash.keys)
  assert_equal(expected_keys_v1,hash['v1'].keys.sort)
  assert_equal(expected_keys_v1,hash['v1'].values.sort)
end

# File pkg/statsample-1.4.0/test/test_codification.rb, line 32
def test_create_yaml
  assert_raise  ArgumentError do
    Statsample::Codification.create_yaml(@ds,[])
  end
  expected_keys_v1=%w{run walk walking running sleep sleeping dream dreaming}.sort
  yaml_hash=Statsample::Codification.create_yaml(@ds,['v1'])
  h=YAML::load(yaml_hash)
  assert_equal(['v1'],h.keys)
  assert_equal(expected_keys_v1,h['v1'].keys.sort)
  tf = Tempfile.new("test_codification")
  yaml_hash=Statsample::Codification.create_yaml(@ds,['v1'],tf, Statsample::SPLIT_TOKEN)
  tf.close
  tf.open
  h=YAML::load(tf)
  assert_equal(['v1'],h.keys)
  assert_equal(expected_keys_v1,h['v1'].keys.sort)
  tf.close(true)
end

# File pkg/statsample-1.4.0/test/test_codification.rb, line 57
def test_recode_dataset_simple
  Statsample::Codification.recode_dataset_simple!(@ds,{'v1'=>@dict})
  expected_vector=['r','w,r','w','r','s','s,d', 's,d'].to_vector
  assert_not_equal(expected_vector,@ds['v1'])
  assert_equal(expected_vector,@ds['v1_recoded'])
end

# File pkg/statsample-1.4.0/test/test_codification.rb, line 63
def test_recode_dataset_split
  Statsample::Codification.recode_dataset_split!(@ds,{'v1'=>@dict})
  e={}
  e['r']=[1,1,0,1,0,0,0].to_vector
  e['w']=[0,1,1,0,0,0,0].to_vector
  e['s']=[0,0,0,0,1,1,1].to_vector
  e['d']=[0,0,0,0,0,1,1].to_vector
  e.each{|k,expected|
    assert_equal(expected,@ds['v1_'+k],"Error on key #{k}")

  }
end

# File pkg/statsample-1.4.0/test/test_codification.rb, line 50
def test_recodification
  expected=[['r'],['w','r'],['w'],['r'],['s'],['s','d'], ['s','d']]
  assert_equal(expected,Statsample::Codification.recode_vector(@ds['v1'],@dict))
  v2=['run','walk,dreaming',nil,'walk,dream,dreaming,walking'].to_vector
  expected=[['r'],['w','d'],nil,['w','d']]
  assert_equal(expected,Statsample::Codification.recode_vector(v2,@dict))
end