<python>unnitest

最新推荐文章于 2024-03-12 19:33:06 发布

__JoyWang

最新推荐文章于 2024-03-12 19:33:06 发布

阅读量453

点赞数

分类专栏： Python

本文链接：https://blog.csdn.net/vulnerableyears/article/details/90489619

版权

Python 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

这篇博客主要介绍如何使用Python内置的unittest模块进行单元测试，通过实例详细讲解了unittest的主要功能和用法，包括测试套件的创建、断言方法的使用以及测试结果的输出。

摘要由CSDN通过智能技术生成

main function：进行unnitest

import unittest
import shopping_cart

def prepare_cart():
    cart = []

    item = {
        'name': 'Super Awesome Water Bottle',
        'price': 22.95,
        'quantity': 1,
        'sku': 'ABC123'
    }

    item2 = {
        'name': 'Awesome Water Bottle',
        'price': 12.95,
        'quantity': 1,
        'sku': '123DEF'
    }

    shopping_cart.add_item(cart, item)
    shopping_cart.add_item(cart, item2)

    return cart

class TestShoppingCart(unittest.TestCase):
    def test_add_item_to_cart(self):
        cart = []
        item = {
            'name': 'Super Awesome Water Bottle',
            'price': 22.95,
            'quantity': 1,
            'sku': 'ABC123'
        }
        shopping_cart.add_item(cart, item)
        self.assertEqual(len(cart), 1)
        self.assertEqual(cart[0]['sku'], 'ABC123')
        self.assertEqual(cart[0]['price'], 22.95)
        self.assertEqual(cart[0]['name'], 'Super Awesome Water Bottle')

        item2 = {
            'name': 'Awesome Water Bottle',
            'price': 12.95,
            'quantity': 1,
            'sku': '123DEF'
        }

        shopping_cart.add_item(cart, item2)
        self.assertEqual(len(cart), 2)
        self.assertEqual(cart[1]['sku'], '123DEF')
        self.assertEqual(cart[1]['price'], 12.95)
        self.assertEqual(cart[1]['name'], 'Awesome Water Bottle')

    def test_calculate_cart_total(self):
        cart = prepare_cart()
        total = shopping_cart.calculate_total(cart)
        self.assertEqual(total, 35.90)

    def test_remove_item(self):
        cart = prepare_cart()
        shopping_cart.remove_item(cart, 'Super Awesome Water Bottle')
        self.assertEqual(len(cart), 1)
        self.assertEqual(cart[0]['name'], 'Awesome Water Bottle')
        self.assertEqual(cart[0]['price'], 12.95)

    def test_update_quantity(self):
        cart = prepare_cart()
        shopping_cart.update_quantity(cart, 'Super Awesome Water Bottle', 3)
        self.assertEqual(cart[0]['quantity'], 3)
        total = shopping_cart.calculate_total(cart)
        self.assertEqual(total, 81.80)

    def test_set_quantity_zero(self):
        cart = prepare_cart()
        shopping_cart.update_quantity(cart, 'Super Awesome Water Bottle', 0)
        self.assertEqual(len(cart), 1)
        total = shopping_cart.calculate_total(cart)
        self.assertEqual(total, 12.95)
        self.assertEqual(cart[0]['name'], 'Awesome Water Bottle')


    def test_set_quantity_negative(self):
        cart = prepare_cart()
        shopping_cart.update_quantity(cart, 'Awesome Water Bottle', -1)
        self.assertEqual(len(cart), 1)
        total = shopping_cart.calculate_total(cart)
        self.assertEqual(total, 22.95)
        self.assertEqual(cart[0]['name'], 'Super Awesome Water Bottle')




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

module shopping_cart: 被检验的自定义函数

def remove_item(cart, name):
    for i in range(len(cart)):
        if cart[i]['name'] == name:
            del cart[i]
            break

def calculate_total(cart):
    total = 0
    for item in cart:
        total += item['price'] * item['quantity']
    return total


def add_item(cart, item):
    cart.append(item)


def update_quantity(cart, name, quantity):
    for item in cart:
        if item['name'] == name:
            item['quantity'] = quantity

    if quantity <= 0:
        remove_item(cart, name)


def main():
    cart = []
    item = {
        'name': 'Super Awesome Water Bottle',
        'price': 22.95,
        'quantity': 1,
        'sku': 'ABC123'
    }

    item2 = {
        'name': 'Awesome Water Bottle',
        'price': 12.95,
        'quantity': 1,
        'sku': '123DEF'
    }

    add_item(cart, item)
    add_item(cart, item2)
    remove_item(cart, 'Super Awesome Water Bottle')
    update_quantity(cart, 'Awesome Water Bottle', 3)