def test_missing_encode_features(self):
X_enc, enc_map = encode_features(SPOTTY_CAT)
expected_X = np.array([[0, 0], [0, 0], [0, 0], [1, 0], [-1, 0], [3, 0],
[3, 1], [-1, 0], [2, 0], [2, 0], [0, 0], [0,
2]])
assert_array_equal(X_enc, expected_X)
self.assertEqual(enc_map, [{
0.: 0,
1.: 1,
8.: 2,
9.: 3
}, {
0.: 0,
1.: 1,
2.: 2
}])
def test_encode_features(self):
X_enc, enc_map = encode_features(STOCKS_CAT)
expected_X = np.array([[3, 2], [2, 2], [3, 2], [3, 2], [1, 2], [1, 2],
[4, 0], [0, 2], [0, 2], [4, 2], [3, 2], [2, 1]])
assert_array_equal(X_enc, expected_X)
self.assertEqual(enc_map, [{
'cons': 0,
'fin': 1,
'nrg': 2,
'tech': 3,
'tel': 4
}, {
'CN': 0,
'NL': 1,
'USA': 2
}])
def test_missing_encode_features(self):
X_enc, enc_map = encode_features(SPOTTY_CAT)
expected_X = np.array([[0, 0],
[0, 0],
[0, 0],
[1, 0],
[-1, 0],
[3, 0],
[3, 1],
[-1, 0],
[2, 0],
[2, 0],
[0, 0],
[0, 2]])
assert_array_equal(X_enc, expected_X)
self.assertEqual(enc_map,
[{0.: 0, 1.: 1, 8.: 2, 9.: 3},
{0.: 0, 1.: 1, 2.: 2}])
def test_encode_features(self):
X_enc, enc_map = encode_features(STOCKS_CAT)
expected_X = np.array([[3, 2],
[2, 2],
[3, 2],
[3, 2],
[1, 2],
[1, 2],
[4, 0],
[0, 2],
[0, 2],
[4, 2],
[3, 2],
[2, 1]])
assert_array_equal(X_enc, expected_X)
self.assertEqual(enc_map,
[{'cons': 0, 'fin': 1, 'nrg': 2, 'tech': 3, 'tel': 4},
{'CN': 0, 'NL': 1, 'USA': 2}])
