11# encoding=utf-8
22from collections import OrderedDict
33import unittest
4- from mock import Mock
4+ import unittest . mock as mock
55from unittest import TestCase
66
77from faker import Faker
@@ -19,31 +19,31 @@ class DummyDataProvider:
1919
2020 def page_mock (self , name = None , site = None , prefix = '' ):
2121 """ Create a MWClient Page instance mock """
22- page = Mock (Page )
22+ page = mock . Mock (Page )
2323 page .site = site or self .site
2424 page .name = '%s%s' % (prefix , name or fake .name ())
2525 return page
2626
2727 def article_mock (self , name = None , site = None ):
2828 """ Create an UKBOt Article instance mock """
29- article = Mock (Article )
30- article .site = Mock (return_value = site or self .site )
29+ article = mock . Mock (Article )
30+ article .site = mock . Mock (return_value = site or self .site )
3131 article .name = name or fake .name ()
3232 article .key = article .site ().key + ':' + article .name
3333 return article
3434
3535 def __init__ (self , articles : int , categories : int ):
3636
3737 # Define a dummy site
38- self .site = Mock (Site )
38+ self .site = mock . Mock (Site )
3939 self .site .key = 'dummy.wikipedia.org'
4040 # self.site.redirect_regexp = re.compile(u'(?:%s)' % u'|'.join(['redirect']), re.I)
4141 self .site .rights = ['bot' ]
4242
4343 # And a site manager
44- self .sites = Mock (SiteManager )
45- self .sites .keys = Mock (return_value = [self .site .key ])
46- self .sites .items = Mock (return_value = [(self .site .key , self .site )])
44+ self .sites = mock . Mock (SiteManager )
45+ self .sites .keys = mock . Mock (return_value = [self .site .key ])
46+ self .sites .items = mock . Mock (return_value = [(self .site .key , self .site )])
4747
4848 # Create some articles and categories
4949 self .articles = [self .article_mock () for n in range (articles )]
0 commit comments