def test_predict_dense_cb_explore_adf(self):
actual = VowpalLearner._predict_format(True, (1, 2, 3), [(1, 0, 0),
(0, 1, 0),
(0, 0, 1)])
expected = "shared |s 0:1 1:2 2:3\n|a 0:1\n|a 1:1\n|a 2:1"
self.assertEqual(actual, expected)
def test_predict_sparse_cb_explore_adf_2(self):
actual = VowpalLearner._predict_format(True, {
"D": 1,
"C": 2
}, ["A", "B", "C"])
expected = "shared |s D:1 C:2\n|a A:1\n|a B:1\n|a C:1"
self.assertEqual(actual, expected)
def test_predict_sparse_cb_explore_1(self):
actual = VowpalLearner._predict_format(False, {
'a': 1,
3: 2
}, ["A", "B", "C"])
expected = "|s a:1 3:2"
self.assertEqual(actual, expected)
def test_predict_sparse_cb_explore_adf_3(self):
actual = VowpalLearner._predict_format(True, ((1, 3), (1, 2)),
["A", "B", "C"])
expected = "shared |s 1:1 3:2\n|a A:1\n|a B:1\n|a C:1"
self.assertEqual(actual, expected)
def test_predict_sparse_cb_explore_2(self):
actual = VowpalLearner._predict_format(False, ((1, 5, 7), (2, 2, 3)),
["A", "B", "C"])
expected = "|s 1:2 5:2 7:3"
self.assertEqual(actual, expected)
def test_predict_dense_cb_explore(self):
actual = VowpalLearner._predict_format(False, (1, 2, 3),
["A", "B", "C"])
expected = "|s 0:1 1:2 2:3"
self.assertEqual(actual, expected)
