示例#1
0
def testLogsToSeti():
  # First three lines are copied from tdg_v0.csv
  csvs = ['testdata/logs_to_seti_csv0.csv']
  # s0. dob: 05/06/1989. gender:f
  # s1. dob: 05/06/1949. gender:m
  s0 = seti.create_seti(19.15, bfs=[('has_bite_dog', 'N')], cfs=[('dob', 26.0)])
  s1 = seti.create_seti(22.51, bfs=[('has_bite_dog', 'N')], cfs=[('dob', 66.0)])

  # Test model gets created and loaded.
  # Test that we can score one example.
  setis = logs_to_seti.generate_seti(csvs, for_test=True)
  wants = [s0, s1]
  for i in xrange(len(setis)):
    assertEquals(str(wants[i]), str(setis[i]))