"""
Exercise 1.6 - Units and Tens
"""


#  Copyright (c) 2021. Pascal Syma. All rights reserved.

def unit(num: int) -> int:
    """
    Return the 'ones' digit.

    :return: Ones digit
    :param num: Input number
    """

    return int(str(num)[-1:])


def ten(num: int) -> int:
    """
    Return the 'tens' digit.

    :return: Tens digit
    :param num: Input number
    """

    return int(str(num)[-2:-1])


if __name__ == '__main__':
    print(f'{unit(123) = }')
    print(f'{ten(123) = }')