虫师 large avatar
测试学习小组  ›  python selenium

python练习题1:检查文件后缀名,请实现一个函数,使测试用例通过!

By 虫师 at 8 个月前 , 704 次浏览
import unittest



def yaml_to_json(file_name):
# 请在这里实现代码
pass


class MyTest(unittest.TestCase):

def test_01(self):
result = yaml_to_json("abc.abc")
self.assertEqual(result, "文件格式错误")

def test_02(self):
result = yaml_to_json("abc.yaml")
self.assertEqual(result, "abc.json")

def test_03(self):
result = yaml_to_json("abc.abc.yaml.yaml")
self.assertEqual(result, "abc.abc.yaml.json")

def test_04(self):
result = yaml_to_json("yaml.yaml")
self.assertEqual(result, "yaml.json")

def test_05(self):
result = yaml_to_json("yaml")
self.assertEqual(result, "文件没有后缀名")

def test_06(self):
result = yaml_to_json("123.YAML")
self.assertEqual(result, "123.json")

def test_07(self):
result = yaml_to_json("abc.yaml.lower()")
self.assertEqual(result, "文件格式错误")

if __name__ == '__main__':
unittest.main()


4 回复
heimao0307 medium avatar
#1 - 7 个月前
heimao0307
import unittest
import os



def yaml_to_json(file_name):
# 使用splitext函数取出文件扩展名
ext = os.path.splitext(file_name)[1]
if ext in (".yaml", ".YAML"):
return os.path.splitext(file_name)[0] + ".json"
elif ext == "":
return "文件没有后缀名"
else:
return "文件格式错误"


乙醇 medium avatar
#2 - 6 个月前
乙醇

我给你改了个格式

煎饼果子 medium avatar
#3 - 6 个月前
煎饼果子
def yaml_to_json(file_name):
# 请在这里实现代码
array4FileName = file_name.split('.')
if len(array4FileName)<2:
return "文件没有后缀名"
elif array4FileName[-1] !="yaml" and array4FileName[-1] !="YAML":
return "文件格式错误"
else:
array4FileName[-1] = 'json'
return '.'.join(array4FileName)

LZL medium avatar
#4 - 6 个月前
LZL
import unittest
import os

def yaml_to_json(file_name):
# 请在这里实现代码
names = os.path.splitext(file_name)
expande_name = names[1]
if expande_name == '.yaml' or expande_name == '.YAML':
return names[0] + '.json'
elif str(expande_name).strip() == '':
return "文件没有后缀名"
return '文件格式错误'

测试学习小组 — 软件测试学习讨论区
现在注册
已注册用户请 登入
内容分享

欢迎关注我们的公众号: 测试圈TC


小贴士

你的问题在测试教程网中能找到答案