def test_get_extension(self): mf = MediaFile(type = Type(code='video'), \ status = Status(code='rename'), \ filepath ='/fakefilepath/myfile.mkv') self.assertEqual(mf.get_extension(), 'mkv') mf.filepath = '/fakefilepath/myfile.test.mkv' self.assertEqual(mf.get_extension(), 'mkv')
def test_get_extension(self): mf = MediaFile(type = Type(code='video'), \ status = Status(code='rename'), \ filepath ='/fakefilepath/myfile.mkv') self.assertEqual(mf.get_extension(), 'mkv') mf.filepath ='/fakefilepath/myfile.test.mkv' self.assertEqual(mf.get_extension(), 'mkv')