| 28 | ids=["int64", "float32-nan", "object", "object-None", "str"], |
| 29 | ) |
| 30 | def test_encode_util(values, expected): |
| 31 | uniques = _unique(values) |
| 32 | assert_array_equal(uniques, expected) |
| 33 | |
| 34 | result, encoded = _unique(values, return_inverse=True) |
| 35 | assert_array_equal(result, expected) |
| 36 | assert_array_equal(encoded, np.array([1, 0, 2, 0, 2])) |
| 37 | |
| 38 | encoded = _encode(values, uniques=uniques) |
| 39 | assert_array_equal(encoded, np.array([1, 0, 2, 0, 2])) |
| 40 | |
| 41 | result, counts = _unique(values, return_counts=True) |
| 42 | assert_array_equal(result, expected) |
| 43 | assert_array_equal(counts, np.array([2, 1, 2])) |
| 44 | |
| 45 | result, encoded, counts = _unique(values, return_inverse=True, return_counts=True) |
| 46 | assert_array_equal(result, expected) |
| 47 | assert_array_equal(encoded, np.array([1, 0, 2, 0, 2])) |
| 48 | assert_array_equal(counts, np.array([2, 1, 2])) |
| 49 | |
| 50 | |
| 51 | def test_encode_with_check_unknown(): |