Modo corretto per affermare il tipo di variabile in Python

Question

Nell'utilizzo di una funzione, desidero garantire che il tipo di variabili sia come previsto. Come farlo bene?

Qui è una funzione di falso esempio cercando di fare proprio questo prima di andare avanti con il suo ruolo:

def my_print(begin, text, end): 
    """Print 'text' in UPPER between 'begin' and 'end' in lower 

    """ 
    for i in (begin, text, end): 
     assert isinstance(i, str), "Input variables should be strings" 
    out = begin.lower() + text.upper() + end.lower() 
    print out 

def test(): 
    """Put your test cases here! 

    """ 
    assert my_print("asdf", "fssfpoie", "fsodf") 
    assert not my_print("fasdf", 33, "adfas") 
    print "All tests passed" 

test() 

È affermare l'approccio giusto? Dovrei usare try/except invece?

Inoltre, la mia asserzione serie di test non sembra funzionare correttamente: S

Grazie pythoneers

Answer 1

Il isinstance integrato è il modo preferito se proprio devi, ma ancora meglio è ricordare il motto di Python: "è più facile chiedere perdono che permesso"! -) (In realtà era il motto preferito di Grace Murray Hopper;). Vale a dire:!

def my_print(text, begin, end): 
    "Print 'text' in UPPER between 'begin' and 'end' in lower" 
    try: 
     print begin.lower() + text.upper() + end.lower() 
    except (AttributeError, TypeError): 
     raise AssertionError('Input variables should be strings') 

questo, BTW, lascia il lavoro di funzione bene su stringhe Unicode -, senza alcuno sforzo aggiuntivo -)

Answer 2

Si potrebbe desiderare di provare questo esempio per la versione 2.6 di Python.

def my_print(text, begin, end): 
    "Print text in UPPER between 'begin' and 'end' in lower." 
    for obj in (text, begin, end): 
     assert isinstance(obj, str), 'Argument of wrong type!' 
    print begin.lower() + begin.upper() + end.lower() 

Tuttavia, hai considerato di lasciare che la funzione fallisca in modo naturale, invece?

Answer 3

Facendo type('') è effettivamente equivalente a str e types.StringType

così type('') == str == types.StringType valuterà a "True"

Si noti che le stringhe Unicode che contengono solo ASCII falliranno se si controllano i tipi in questo modo, quindi si consiglia di fare somethi ng come assert type(s) in (str, unicode) o assert isinstance(obj, basestring), l'ultimo dei quali è stato suggerito nei commenti da 007Brendan ed è probabilmente preferito.

isinstance() è utile se si vuole chiedere se un oggetto è un'istanza di una classe, per esempio:

class MyClass: pass 

print isinstance(MyClass(), MyClass) # -> True 
print isinstance(MyClass, MyClass()) # -> TypeError exception 

Ma per i tipi di base, ad esempio, str, unicode, int, float, long ecc. Chiedendo type(var) == TYPE funzionerà OK.