2015-10-05 102 views
1

我做了一个连接到twitter api的函数。 该函数返回一个twitter对象。我想创建一个测试函数来检查返回的对象是否真的是一个Twitter对象。 所以这是我的函数:如何测试python中两个对象是否相等?

def authenticate_twitter_api(): 
    """Make connection with twitters REST api""" 
    try: 
     logger.info('Starting Twitter Authentication') 
     twitter_api = twitter.Twitter(auth=twitter.OAuth(config.TWITTER_ACCESS_KEY, config.TWITTER_ACCESS_SECRET, 
                config.TWITTER_CONSUMER_KEY, config.TWITTER_CONSUMER_SECRET)) 
     print twitter_api 
     logger.info("Service has started") 
     return twitter_api 
    except: 
     logger.error("Authentication Error. Could not connect to twitter api service") 

当我运行这个函数返回的是:

<twitter.api.Twitter object at 0x7fc751783910> 

现在,我想创建一个测试功能,也许通过numpy.testing为了检查类型是一个对象。

numpy.testing.assert_equal(actual, desired, err_msg='') 

actual = type(authenticate_twitter_api()) 
desired =<class 'twitter.api.Twitter'> 

这就是问题所在。我无法将对象保存为“所需”。

我该怎么办?

回答

2

您正在查找的所需对象是twitter.api.Twitter,只需将其导入并传递类assert_equal即可。

然而,这是更地道使用isinstance

from twitter.api import Twitter 

if isinstance(authenticate_twitter_api(), Twitter): 
    print("It was a Twitter object.") 
+0

完美工作。谢谢! –

2

类对象本身在Python。所以你可以像这样分配你想要的变量:

import twitter 
# (...) 
desired = twitter.api.Twitter 
相关问题