test/test.rb

require 'predictionio'
require 'date'
require 'test/unit'

class TestPredictionIO < MiniTest::Unit::TestCase
  if ENV["APPKEY"] then
    APPKEY = ENV["APPKEY"]
  else
    APPKEY = "k4f6rCV8YTM5x0PbRdIG4yrWiKLhOv16V0Q8COE2AcvnYmSlxbAcXR5pucI5HO21"
  end
  if ENV["APIURL"] then
    APIURL = ENV["APIURL"]
  else
    APIURL = "http://localhost:8000"
  end
  APITHREADS = 1

  def test_appkey
    client = PredictionIO::Client.new("foobar", APITHREADS, APIURL)
    assert_equal(client.appkey, "foobar")
  end

  def test_get_status
    client = PredictionIO::Client.new("foobar", APITHREADS, APIURL)
    assert_equal("PredictionIO Output API is online.", client.get_status)
  end

  def test_user
    client = PredictionIO::Client.new(APPKEY, APITHREADS, APIURL)
    client.create_user("ruby_foobar")
    assert_equal("ruby_foobar", client.get_user("ruby_foobar")["pio_uid"], "uid: ruby_foobar")
    client.delete_user("ruby_foobar")
    client.create_user("ruby_barbaz",
                       "gender" => "F",
                       "bday" => "1985-05-05",
                       "pio_latitude" => 21.109,
                       "pio_longitude" => -48.7479,
                       "pio_inactive" => true)
    ruby_barbaz = client.get_user("ruby_barbaz")
    assert_equal("ruby_barbaz", ruby_barbaz["pio_uid"], "uid: ruby_barbaz")
    #assert_equal("F", ruby_barbaz["gender"], "gender: F")
    assert_equal(21.109, ruby_barbaz["pio_latitude"], "lat: 21.109")
    assert_equal(-48.7479, ruby_barbaz["pio_longitude"], "lng: -48.7479")
    #assert_equal("1985-05-05", ruby_barbaz["bday"], "bday: 1985-05-05")
    assert(ruby_barbaz["pio_inactive"], "inactive: true")
    client.delete_user("ruby_barbaz")
  end

  def test_item
    client = PredictionIO::Client.new(APPKEY, APITHREADS, APIURL)
    client.create_item("ruby_barbaz",
                       ["218", "55"],
                       "pio_latitude" => -58.24089,
                       "pio_longitude" => 48.17890,
                       "pio_startT" => Time.at(478308922000))
    assert_raises(PredictionIO::Client::ItemNotFoundError) { client.get_item("randomstuff") }
    ruby_barbaz = client.get_item("ruby_barbaz")
    assert_equal("ruby_barbaz", ruby_barbaz["pio_iid"], "iid: ruby_barbaz")
    assert(ruby_barbaz["pio_itypes"].include?("218"), "itypes: 218")
    assert(ruby_barbaz["pio_itypes"].include?("55"), "itypes: 55")
    assert_equal(-58.24089, ruby_barbaz["pio_latitude"], "lat: -58.24089")
    assert_equal(48.1789, ruby_barbaz["pio_longitude"], "lng: 48.1789")
    assert_equal(Time.at(478308922000), ruby_barbaz["pio_startT"], "startT: 478308922000")
    client.delete_item("ruby_barbaz")
  end

  def test_u2i
    client = PredictionIO::Client.new(APPKEY, APITHREADS, APIURL)
    client.identify("foo1")
    client.record_action_on_item("rate", "bar2", "pio_rate" => 4)
    client.identify("foo2")
    client.record_action_on_item("like", "bar4")
    client.identify("foo4")
    client.record_action_on_item("dislike", "bar8")
    client.identify("foo8")
    client.record_action_on_item("view", "bar16")
    client.identify("foo16")
    client.record_action_on_item("conversion", "bar32")
  end

  def test_itemrec
    client = PredictionIO::Client.new(APPKEY, APITHREADS, APIURL)
    client.identify("218")
    iids = client.get_itemrec_top_n("test", 5)
    assert(iids.include?("itemrec"))
    assert(iids.include?("218"))
    assert(iids.include?("1"))
    assert(iids.include?("foo"))
    assert(iids.include?("bar"))
    assert_equal(iids.length, 5)
  end
end