def setUp(self):
self.replace = Replacer()
self.addCleanup(self.replace.restore)
self.server = MockOpen('https://test.attask-ondemand.com/attask/api/' +
self.api_version)
self.replace('workfront.six.moves.urllib.request.urlopen', self.server)
self.session = Session('test', api_version=self.api_version)
self.api = self.session.api
def test_load_data():
with mock.patch("builtins.open", MockOpen(string_data)) as open_mock:
points = ukpolice.load("filename", {"Burglary"})
assert( open_mock.calls[0][0] == ("filename",) )
assert( len(points.timestamps) == 2 )
assert( points.timestamps[0] == np.datetime64("2016-03") )
assert( points.timestamps[1] == np.datetime64("2017-01") )
np.testing.assert_allclose( points.coords[:,0], np.array([123, 456]) )
np.testing.assert_allclose( points.coords[:,1], np.array([789, 1012]) )
def set_up(test):
test.replace = Replacer()
test.addCleanup(test.replace.restore)
test.server = MockOpen(
'https://test.attask-ondemand.com/attask/api/unsupported')
test.replace('workfront.six.moves.urllib.request.urlopen', test.server)
filename = os.path.split(str(test))[-1]
key = os.path.splitext(filename)[0]
for response in RESPONSES.get(key, ()):
test.server.add(**response)
def setUp(self):
r = Replacer()
self.addCleanup(r.restore)
self.mock = Mock()
self.mock.input.return_value = 'u'
self.mock.getpass.return_value = 'p'
r.replace('workfront.get_api_key.input', self.mock.input)
r.replace('getpass.getpass', self.mock.getpass)
r.replace('logging.basicConfig', self.mock.logging)
r.replace('sys.argv', ['x'])
self.server = MockOpen(
'https://api-cl01.attask-ondemand.com/attask/api/unsupported')
r.replace('workfront.six.moves.urllib.request.urlopen', self.server)
def test_generated_methods(self):
for api_version, type_, name in self.methods_to_test():
session = Session('test', api_version=api_version)
obj = type_(session, ID='xxx')
method = getattr(obj, name)
workfront_name = re.search('``(.+?)``', method.__doc__).group(1)
server = MockOpen('https://test.attask-ondemand.com/attask/api/' +
api_version)
server.add(url='/{}/xxx/{}'.format(type_.code, workfront_name),
params='method=PUT',
response='{"data": {"result": null}}')
with Replace('workfront.six.moves.urllib.request.urlopen', server):
method()
def test_load_default_filename():
with mock.patch("builtins.open", MockOpen(None)) as open_mock:
assert( ukpolice.default_burglary_data() == None )
filename = open_mock.calls[0][0][0]
assert( os.path.split(filename)[1] == "uk_police.csv" )
