[lang-ref] ( get_extension ) ( python )

def test_get_extension():
    # os.path.splitext(path)[1]
    assert os.path.splitext('file1.txt')[1]          == '.txt'  # Note: includes dot
    assert os.path.splitext('file1.tar.gz')[1]       == '.gz'   # Note: Last ext only
    assert os.path.splitext('.bashrc')[1]            == ''      # Note: . prefix is not assumed as ext
    assert os.path.splitext('/path/to/file1.txt')[1] == '.txt'

def test_get_extension_alternative():
    # pathlib.Path.suffix
    p1 = Path('file1.txt')
    assert p1.suffix == '.txt'

    p2 = Path('file1.tar.gz')
    assert p2.suffix == '.gz'
    assert p2.suffixes == ['.tar', '.gz']

    p3 = Path('.bashrc')
    assert p3.suffix == ''

    p4 = Path('/path/to/file1.txt')
    assert p4.suffix  == '.txt'