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}])