Python - 使用多種類型作爲輸入定義函數

Question

假設我試圖在python中創建一個假想的數字類（就像一個練習，我知道那裏已經有一個內置的）。您應該可以將虛數加到其他虛數以及虛數上，反之亦然。

我得到了使用__add__()向其他虛構中添加虛構的一個掛鉤，我似乎可以使用鴨子打字將虛構添加到整數（儘管我不太明白），但我不知道如何給虛構添加一個int。請發送幫助。

Answer 1

對於添加你定義添加方法。

def __add__(self, other): 

，並在你試圖識別其他參數的類型：

if isinstance(other, imaginary): 

您執行基於這些測試的結果計算。

更多的是在這裏解釋：

class ComplexTest(unittest.TestCase): 

    def assertAlmostEqual(self, a, b): 
     if isinstance(a, complex): 
      if isinstance(b, complex): 
       unittest.TestCase.assertAlmostEqual(self, a.real, b.real) 
       unittest.TestCase.assertAlmostEqual(self, a.imag, b.imag) 
      else: 
       unittest.TestCase.assertAlmostEqual(self, a.real, b) 
       unittest.TestCase.assertAlmostEqual(self, a.imag, 0.) 
     else: 
      if isinstance(b, complex): 
       unittest.TestCase.assertAlmostEqual(self, a, b.real) 
       unittest.TestCase.assertAlmostEqual(self, 0., b.imag) 
      else: 
       unittest.TestCase.assertAlmostEqual(self, a, b) 

Answer 2

實施__add__任何東西添加到您的類並實現__radd__實現添加：https://docs.python.org/2/library/numbers.html#implementing-the-arithmetic-operations

您也可以從test_complex.py測試用例的想法你的班級實例到其他地方。在內置類型上使用isinstance並不是一個可怕的想法。在你自己的課堂上使用它通常是一個匆忙或不好的設計的標誌。但是如果你確定類的列表不會太多，不要擔心。如果它會增長，請考慮使用訪問者模式重新設計。

從Python data model documentation：

對於對象x和y，第一x.__op__(y)試圖。如果未實施或返回NotImplemented，則嘗試使用y.__rop__(x)。如果這還沒有執行或者返回NotImplemented，則會引發TypeError異常。