def create_one_high_tree(filename=father_filename):
sample_serializer = serializer.Serializer()
daughter_node = imagetree.ImageTree(name='daughter',
children_links=[],
input_image=(),
serializer=sample_serializer,
children=[],
filename=filename)
daughter_node2 = imagetree.ImageTree(name='daughter2',
children_links=[],
input_image=(10, 50, 100),
serializer=sample_serializer,
children=[],
filename=filename)
son_node = imagetree.ImageTree(name='son',
children_links=[],
input_image=TestImageTree.son_pixel,
children=[],
filename=filename,
serializer=sample_serializer)
son_node2 = imagetree.ImageTree(name='son2',
children_links=[],
input_image=(20, 52, 200),
children=[],
filename=filename,
serializer=sample_serializer)
father_node = imagetree.ImageTree(
name='father',
children_links=copy.deepcopy(TestImageTree.children_names),
input_image=TestImageTree.father_pixel,
children=[son_node, daughter_node, son_node2, daughter_node2],
serializer=sample_serializer,
filename=filename)
return father_node
def test_get_child_leaf(self):
leaf = imagetree.ImageTree(children=[],
name='test_leaf',
input_image=(),
children_links=[],
serializer=serializer.Serializer(),
filename='')
self.assertTrue(leaf.is_leaf())
self.assertEqual(1, leaf.height(max_height=999))
self.assertEqual(1, leaf.count_nodes())
def create_image_url_tree(self):
first_serialized_string = 'first\t\t\t\tchild0\tchild1\tchild2\tchild3\t' + ImageTests.wiki_image_url + '\n'
child0_serialized_string = 'child0\t\t\t\t\t\t\t\t' + ImageTests.wiki_image_url + '\n'
child1_serialized_string = 'child1\t\t\t\t\t\t\t\t' + '\n'
child2_serialized_string = 'child2\t\t\t\t\t\t\t\t' + '\n'
child3_serialized_string = 'child3\t\t\t\t\t\t\t\t' + '\n'
full_serialized_string = first_serialized_string + child0_serialized_string + child1_serialized_string \
+ child2_serialized_string + child3_serialized_string
filename = 'deserialize_image_url.tsv'
test_serializer = serializer.Serializer()
deserialized_tree = test_serializer.load_from_string(
node_name='first',
filename=filename,
serialized_string=full_serialized_string)
return deserialized_tree
def test_insert_tree(self):
ser = serializer.Serializer()
first_tree = serializer.create_tree_from_jpg_url(
ImageTests.wiki_image_url,
name='test_insert',
filename='test_insert',
serializer=ser)
second_tree = serializer.create_tree_from_jpg_url(
ImageTests.wiki_image_url,
name='child',
filename='test_insert',
serializer=ser)
test_quadkey = '012'
first_tree.insert(another_tree=second_tree, quad_key=test_quadkey)
rendered = first_tree.get_np_array(128)
self.assertEqual(rendered.shape[0], 128)
def test_consistent_image_as_you_zoom(self):
gm = graphmap_main.GraphMap(serializer.Serializer())
fruits_node_link = NodeLink(constants.FRUITS_LINK)
lowest_quad_keys = ['333133', '312312', '131231']
for lowest_quad_key in lowest_quad_keys:
expected_same_images = []
for i in range(len(lowest_quad_key) + 1):
node_quad_key = lowest_quad_key[:i]
resolution = 2**(len(lowest_quad_key) - i) * 32
node_pil_image_result = gm.get_image_at_quad_key(
root_node_link=fruits_node_link,
resolution=resolution,
quad_key=node_quad_key)
self.assertTrue(node_pil_image_result.is_success())
node_pil_image = node_pil_image_result.value
image_quad_key = lowest_quad_key[i:]
expected_same_image = imagetree.pil_image_at_quad_key(
node_pil_image, quad_key=image_quad_key)
expected_same_images.append(expected_same_image)
for expected_same_image in expected_same_images[1:]:
self.assertTrue(
utilities.pil_images_equal(expected_same_image,
expected_same_images[0]))
def test_picture_in_picture_render(self):
jpg_web_image = imagevalue.JpgWebImage(ImageTests.wiki_image_url)
ser = serializer.Serializer()
child_tree = imagetree.ImageTree(name='child_w',
children_links=[],
filename='',
serializer=ser,
input_image=jpg_web_image)
children = [
child_tree,
imagetree.empty_image_tree('1', ser, ''),
imagetree.empty_image_tree('2', ser, ''),
imagetree.empty_image_tree('3', ser, '')
]
top_tree = imagetree.ImageTree(name='father_w',
children_links=[],
filename='',
serializer=ser,
input_image=jpg_web_image,
children=children)
rendered = top_tree.get_pil_image(128)
self.assertEqual(rendered.size, (128, 128))