curriculum/challenges/english/blocks/daily-coding-challenges-python/68f6587287ad1f4ad39b0c7e.md
Given a string representing a filename, return the extension of the file.
.)."none".get_extension("document.txt") should return "txt".
({test: () => { runPython(`
from unittest import TestCase
TestCase().assertEqual(get_extension("document.txt"), "txt")`)
}})
get_extension("README") should return "none".
({test: () => { runPython(`
from unittest import TestCase
TestCase().assertEqual(get_extension("README"), "none")`)
}})
get_extension("image.PNG") should return "PNG".
({test: () => { runPython(`
from unittest import TestCase
TestCase().assertEqual(get_extension("image.PNG"), "PNG")`)
}})
get_extension(".gitignore") should return "gitignore".
({test: () => { runPython(`
from unittest import TestCase
TestCase().assertEqual(get_extension(".gitignore"), "gitignore")`)
}})
get_extension("archive.tar.gz") should return "gz".
({test: () => { runPython(`
from unittest import TestCase
TestCase().assertEqual(get_extension("archive.tar.gz"), "gz")`)
}})
get_extension("final.draft.") should return "none".
({test: () => { runPython(`
from unittest import TestCase
TestCase().assertEqual(get_extension("final.draft."), "none")`)
}})
def get_extension(filename):
return filename
def get_extension(filename):
last_dot = filename.rfind('.')
if last_dot == -1 or last_dot == len(filename) - 1:
return "none"
return filename[last_dot + 1:]