Question

我试图编写一个简单的堆栈类来了解TDD。但是问题是它无法使用正确的代码通过测试。

代码如下：

class Stack:
    def __init__(self):
        self.stack = []

    def push(self,new_item):
        self.stack.append(new_item)

    def pop(self):
        return int(self.stack.pop(0))

这是测试类：

import pytest
from Stack import Stack

def test_it_can_push():
    stack = Stack()
    stack.push(2)
    assert stack.stack is [2]

这是错误：

    def test_it_can_push():
        stack = Stack()
        stack.push(2)
>       assert stack.stack is [2]
E       assert [2] is [2]
E        +  where [2] = <Stack.Stack instance at 0x7f2273491560>.stack

test_stack.py:7: AssertionError

有人可以告诉我如何解决此问题吗？

Answer 1

您正在使用id进行身份检查（is – CPython中的内存位置），因为操作数是两个不同的列表（它们是可变对象），所以它永远不会相等。它们具有相同的元素，您可以使用id进行检查。

进行股权测试：

assert stack.stack == [2]

如何使用pytest测试一个简单的堆栈类

1 个答案: