def METHOD_NAME(self):
    input_array = np.array([[1, 2, 3, 1], [0, 3, 1, 0]])
    expected_output = np.array([[0, 2, 1, 1, 0, 0], [2, 1, 0, 1, 0, 0]])
    num_tokens = 6
    expected_output_shape = (2, num_tokens)
    layer = layers.CategoryEncoding(num_tokens=6, output_mode="count")
    int_data = layer(input_array)
    self.assertEqual(expected_output_shape, int_data.shape)
    self.assertAllClose(int_data, expected_output)