"""
Tests for creating and storing dominions
"""
import src
from dominion import random_dominion
from graphs import random_tree


print('Create a random dominion of size 4 with labels {0,1,2}.')
D = random_dominion(4, range(3))
print('Show some information about this dominion.')
print(D)
print('Display the dominion we created.')
D.show()
print()

print('We can also demand that only certain labels can appear next to each \
other.')
print('We will use a tree to illustrate this, although we could have used any \
kind of graph we like.')
print('The vertices of our tree will be {0,1,2,3,4,5}.')
T = random_tree(range(6))
print(T)
print('Create a new random dominion of size 10 with labels {0,1,2,3,4,5} and \
display it.')
D = random_dominion(10, range(6), T)
D.show()
print()

print('We can also save an image of this dominion as a file.')
print('We\'ll use `magma` for our color map.')
D.draw('magma', 'dominion_draw_test')
print()

print('Let\' see how long it takes to make a bigger dominion and draw it.')
D = random_dominion(1000, range(6), T)
D.draw('magma', 'dominion_draw_test_big')
print()