32岁零基础学Python量化 第3周(附加篇2):光写代码不算完,还得学会测代码

#接上面def test_store_single_response(language_survey):     """测试单个答案会被妥善地存储"""     language_survey.store_response('English')     assert 'English' in language_survey.responsesdef test_store_three_responses(language_survey):     """测试三个答案会被妥善地存储"""     responses = ['English', 'Spanish', 'Mandarin']     for response in responses:         language_survey.store_response(response)     for response in responses:         assert response in language_survey.responses

from pathlib import Path import jsonprint('==========欢迎使用记账簿===========')class Accountbook:    """一个关于记账账本的实现"""    def __init__(self, filepath = 'user_information.json'):        self.filepath = Path(filepath)        self.data = self.load_data()    def load_data(self):        """加载数据，不存在或损坏时返回空字典"""        if self.filepath.exists():            try:                user_information = json.loads(self.filepath.read_text())            except json.JSONDecodeError:                user_information = {}  #内容错误时重置为空字典        else:            user_information = {}  #文件不存在时新建一个空字典        return user_information    def add_records(self):        """记账功能"""        date = input('请输入日期(如:5/31):').strip()  #strip用于删除用户多输入的空格        if date.lower() == 'q':   #加lower()防止输入为大写的Q            return        records = self.data.get(date,[]) #错误写法：records = user_information.get(date,[]) ，这里user_information是load_data里的局部变量,在本方法内还未定义         while True:                item = input('商品名称(输入q结束当天):').strip()            if item.lower() == 'q':                  break            cost_str = input('花费¥(输入q结束当天):').strip()             if cost_str.lower() == 'q':                break            try:                cost = float(cost_str)            except ValueError:                print('花费请输入数字,本条记录未保存。')  #如果不是数字就会不记录并提醒用户输入正确数据                continue            records.append({'项目':item,'花费':cost})        if records: #只有在有记录时才保存            self.data[date] = records             self.filepath.write_text(json.dumps(self.data,ensure_ascii=False,indent=2))            print('记录已保存。')        else:            print('当天无记录')    def show_records(self):        """读取账本功能"""        #if先判断文件是否存在,存在则读取,不存在则提醒用户先输入        if self.filepath.exists():            try:                contents = self.filepath.read_text()                information = json.loads(contents)                print(information)                return information            except json.JSONDecodeError:                print('文件损坏，无法读取。')                return {} #文件不存在时没有任何返回值，会隐式返回一个None,导致主程序打印None,应返回明确的空字典{}        else:            #文件不存在时没有任何返回值，会隐式返回一个None,导致主程序打印None,应返回明确的空字典{}            print('内容为空,请先记账')            return {}account1 = Accountbook()account1.load_data()while True:    active = input(        '请问是输入还是查看(输入输:i,查看输:o,退出输:q):').strip().lower()    if active == 'i':        account1.add_records()    elif active == 'o':        account1.show_records()    elif active =='q':        break    else:        print('请按要求输入字母')print('========感谢使用记账簿,欢迎下次使用！==========')

def show_records(self):    """读取账本功能（基于内存数据）"""    if self.data:        print(self.data)        return self.data    else:        print('内容为空,请先记账')        return {}

import pytestimport jsonfrom pathlib import Pathfrom accountbook import Accountbook   # 请确保文件名与你的实际模块名一致@pytest.fixturedef book():    test_file = Path("_test_temp.json")    if test_file.exists():        test_file.unlink()    obj = Accountbook(filepath=str(test_file))    return objdef test_new_book_has_empty_data(book):    """测试1：新创建的账本应无数据且未生成文件"""    assert book.data == {}    assert not book.filepath.exists()def test_write_and_read_data(book):    """测试2：预先给定账本数据，判断是否正确写入文件并可重新加载"""    # 给定的测试数据    test_data = {        "6/3": [            {"项目": "苹果", "花费": 5.5},            {"项目": "香蕉", "花费": 3.0}        ],        "6/4": [            {"项目": "可乐", "花费": 2.5}        ]    }    # 直接将数据写入内存属性    book.data = test_data    # 手动调用文件写入（模拟保存操作）    book.filepath.write_text(json.dumps(book.data, ensure_ascii=False, indent=2))    # 重新创建一个新实例，从同一文件加载    new_book = Accountbook(filepath=str(book.filepath))    # 验证加载的数据与原始数据一致    assert new_book.data == test_data