earthengine-api/python/ee/tests/computed_object_test.py
2013-01-31 18:01:19 -08:00

47 lines
1009 B
Python

# Copyright 2012 Google Inc. All Rights Reserved.
"""Test for the ee.image module."""
import json
import unittest
import ee
class ImageTestCase(unittest.TestCase):
def setUp(self):
ee.algorithms._signatures = {
'Image.fakeFunction': {
'returns': 'Map',
'description': 'Fake doc.',
'args': [
{'type': 'Image', 'name': 'image1', 'description': ''},
]
}
}
def testComputedObject(self):
ee.Initialize(None, '')
# Mock out send.
def MockSend(unused_path, unused_params, unused_method='POST'):
return {'value': 1}
ee.data.send_ = MockSend
# Test serialize
image1 = ee.Image(1)
result = image1.fakeFunction()
self.assertEqual(
{'algorithm': 'Image.fakeFunction',
'image1': {'algorithm': 'Constant', 'value': 1}},
json.loads(result.serialize()))
self.assertEqual({'value': 1}, result.getInfo())
if __name__ == '__main__':
unittest.main()