在計算輸出時，如何在python中使用unittest來測試使用set.pop（）的函數

Question

我想測試一個在計算輸出時使用set.pop( ）的函數。

我想按優先級彈出元素（例如：“ a”將在“ b”之前彈出），但僅在運行測試時才彈出。

我曾考慮過更改函數內部的pop方法（以彈出最小值），但這也會在函數正常運行時（而不是在單元測試中）改變函數的行為。

import unittest

def using_set(arr):
    res = []
    s = set(arr)
    while len(s):
        res.append(s.pop())

    return ','.join(res)


class Test(unittest.TestCase):
    def test_using_set(self):
        self.assertEqual('a,b,g', using_set(['b','a','g']))

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

Answer 1

更好的主意是編寫確定性函數。 在這種情況下，這意味着不僅要按pop返回它們的順序使用s的元素，還要遍歷其元素的排序序列。 （反復更改集合效率很低。）

def using_set(arr):
    res = []
    s = set(arr)
    for item in sorted(s):
        res.append(item)
    return ','.join(res)

如果你不能修改using_set ，那么你需要用你想要的東西相一致的方式來操縱它的返回值using_set返回。

self.assertEqual(['a', 'b', 'g'], sorted(using_set(['b', 'a', 'g']).split(,)))

（如果只可以指定未連接的期望值，則排序后重新連接返回值元素是沒有意義的。）