def testmusicfile_getextension(): print "Check: getextension" testFile = MusicFile("Queen", "Greatist Hits", 4, "Bicycle Race", "/tmp/test/derp.mp3") assert testFile.getartist() == "mp3" or colored("[X] Failed: getextension() check", "red") print colored("\t[!] Success: getextension() check", "green")