Python中怎么將一個(gè)類方法變?yōu)槎鄠€(gè)方法

本篇文章給大家分享的是有關(guān)Python中怎么將一個(gè)類方法變?yōu)槎鄠€(gè)方法，小編覺得挺實(shí)用的，因此分享給大家學(xué)習(xí)，希望大家閱讀完這篇文章后可以有所收獲，話不多說(shuō)，跟著小編一起來(lái)看看吧。

1、ddt 如何實(shí)現(xiàn)參數(shù)化？

先回顧一下上篇文章中 ddt 庫(kù)的寫法：

import unittest  from ddt import ddt,data,unpack  @ddt  class MyTest(unittest.TestCase):      @data((3, 1), (-1, 0), (1.2, 1.0))      @unpack      def test(self, first, second):          pass

ddt 可提供 4 個(gè)裝飾器：1 個(gè)加在類上的 @ddt，還有 3 個(gè)加在類方法上的 @data、@unpack 和 @file_data（前文未提及）。

先看看加在類方法上的三個(gè)裝飾器的作用：

# ddt 版本（win）：1.2.1  def data(*values):      global index_len      index_len = len(str(len(values)))      return idata(values)  def idata(iterable):      def wrapper(func):          setattr(func, DATA_ATTR, iterable)          return func      return wrapper  def unpack(func):      setattr(func, UNPACK_ATTR, True)      return func  def file_data(value):      def wrapper(func):          setattr(func, FILE_ATTR, value)          return func      return wrapper

它們的共同作用是在類方法上 setattr() 添加屬性。至于這些屬性在什么時(shí)候使用？下面看看加在類上的 @ddt 裝飾器源碼：

第一層 for 循環(huán)遍歷了所有的類方法，然后是 if/elif 兩條分支，分別對(duì)應(yīng) DATA_ATTR/FILE_ATTR，即對(duì)應(yīng)參數(shù)的兩種來(lái)源：數(shù)據(jù)（@data）和文件（@file_data）。

elif 分支有解析文件的邏輯，之后跟處理數(shù)據(jù)相似，所以我們把它略過(guò)，主要看前面的 if 分支。這部分的邏輯很清晰，主要完成的任務(wù)如下：

•  遍歷類方法的參數(shù)鍵值對(duì)

•  根據(jù)原方法及參數(shù)對(duì)，創(chuàng)建新的方法名

•  獲取原方法的文檔字符串

•  對(duì)元組和列表類型的參數(shù)作解包

•  在測(cè)試類上添加新的測(cè)試方法，并綁定參數(shù)與文檔字符串

分析源碼，可以看出，@data、@unpack 和 @file_data 這三個(gè)裝飾器主要是設(shè)置屬性并傳參，而 @ddt 裝飾器才是核心的處理邏輯。

這種將裝飾器分散（分別加在類與類方法上），再組合使用的方案，很不優(yōu)雅。為什么就不能統(tǒng)一起來(lái)使用呢？后面我們會(huì)分析它的難言之隱，先按下不表，看看其它的實(shí)現(xiàn)方案是怎樣的？

2、parameterized 如何實(shí)現(xiàn)參數(shù)化？

先回顧一下上篇文章中 parameterized 庫(kù)的寫法：

import unittest  from parameterized import parameterized  class MyTest(unittest.TestCase):      @parameterized.expand([(3,1), (-1,0), (1.5,1.0)])      def test_values(self, first, second):          self.assertTrue(first > second)

它提供了一個(gè)裝飾器類 @parameterized，源碼如下（版本 0.7.1），主要做了一些初始的校驗(yàn)和參數(shù)解析，并非我們關(guān)注的重點(diǎn)，略過(guò)。

我們主要關(guān)注這個(gè)裝飾器類的 expand() 方法，它的文檔注釋中寫到：

A "brute force" method of parameterizing test cases. Creates new test cases and injects them into the namespace that the wrapped function is being defined in. Useful for parameterizing tests in subclasses of 'UnitTest', where Nose test generators don't work.

關(guān)鍵的兩個(gè)動(dòng)作是：“creates new test cases（創(chuàng)建新的測(cè)試單元）”和“inject them into the namespace…（注入到原方法的命名空間）”。

關(guān)于第一點(diǎn)，它跟 ddt 是相似的，只是一些命名風(fēng)格上的差異，以及參數(shù)的解析及綁定不同，不值得太關(guān)注。

最不同的則是，怎么令新的測(cè)試方法生效？

parameterized 使用的是一種“注入”的方式：

inspect 是個(gè)功能強(qiáng)大的標(biāo)準(zhǔn)庫(kù)，在此用于獲取程序調(diào)用棧的信息。前三句代碼的目的是取出 f_locals，它的含義是“l(fā)ocal namespace seen by this frame”，此處 f_locals 指的就是類的局部命名空間。

說(shuō)到局部命名空間，你可能會(huì)想到 locals()，但是，我們之前有文章提到過(guò)“l(fā)ocals() 與 globals() 的讀寫問(wèn)題”，locals() 是可讀不可寫的，所以這段代碼才用了 f_locals。

3、pytest 如何實(shí)現(xiàn)參數(shù)化？

按慣例先看看上篇文章中的寫法：

import pytest  @pytest.mark.parametrize("first,second", [(3,1), (-1,0), (1.5,1.0)])  def test_values(first, second):      assert(first > second)

首先看到“mark”，pytest 里內(nèi)置了一些標(biāo)簽，例如 parametrize、timeout、skipif、xfail、tryfirst、trylast 等，還支持用戶自定義的標(biāo)簽，可以設(shè)置執(zhí)行條件、分組篩選執(zhí)行，以及修改原測(cè)試行為等等。

用法也是非常簡(jiǎn)單的，然而，其源碼可復(fù)雜多了。我們這里只關(guān)注 parametrize，先看看核心的一段代碼：

根據(jù)傳入的參數(shù)對(duì)，它復(fù)制了原測(cè)試方法的調(diào)用信息，存入待調(diào)用的列表里。跟前面分析的兩個(gè)庫(kù)不同，它并沒(méi)有在此創(chuàng)建新的測(cè)試方法，而是復(fù)用了已有的方法。在 parametrize() 所屬的 Metafunc 類往上查找，可以追蹤到 _calls 列表的使用位置：

最終是在 Function 類中執(zhí)行：

好玩的是，在這里我們可以看到幾行神注釋……

以上就是Python中怎么將一個(gè)類方法變?yōu)槎鄠€(gè)方法，小編相信有部分知識(shí)點(diǎn)可能是我們?nèi)粘９ぷ鲿?huì)見到或用到的。希望你能通過(guò)這篇文章學(xué)到更多知識(shí)。更多詳情敬請(qǐng)關(guān)注億速云行業(yè)資訊頻道。