def test_create_shuffle_from_identifier(): Key.set_random_seed(1111) Shuffle.set_random_seed(1112) # Empty keep-same shuffle key = Key() original_shuffle = Shuffle(key.get_size(), Shuffle.SHUFFLE_KEEP_SAME) recreated_shuffle = Shuffle.create_shuffle_from_identifier(original_shuffle.get_identifier()) assert original_shuffle.__repr__() == recreated_shuffle.__repr__() # Empty random shuffle key = Key() original_shuffle = Shuffle(key.get_size(), Shuffle.SHUFFLE_RANDOM) recreated_shuffle = Shuffle.create_shuffle_from_identifier(original_shuffle.get_identifier()) assert original_shuffle.__repr__() == recreated_shuffle.__repr__() # Non-empty keep-same shuffle key = Key.create_random_key(32) original_shuffle = Shuffle(key.get_size(), Shuffle.SHUFFLE_KEEP_SAME) recreated_shuffle = Shuffle.create_shuffle_from_identifier(original_shuffle.get_identifier()) assert original_shuffle.__repr__() == recreated_shuffle.__repr__() # Non-empty random shuffle key = Key.create_random_key(32) original_shuffle = Shuffle(key.get_size(), Shuffle.SHUFFLE_RANDOM) recreated_shuffle = Shuffle.create_shuffle_from_identifier(original_shuffle.get_identifier()) assert original_shuffle.__repr__() == recreated_shuffle.__repr__()
def test_create_shuffle_keep_same(): Key.set_random_seed(1111) Shuffle.set_random_seed(1112) # Empty shuffle. key = Key() assert key.__str__() == "" shuffle = Shuffle(key.get_size(), Shuffle.SHUFFLE_KEEP_SAME) assert shuffle.__str__() == "" # Non-empty shuffle. key = Key.create_random_key(8) assert key.__str__() == "00101100" shuffle = Shuffle(key.get_size(), Shuffle.SHUFFLE_KEEP_SAME) assert shuffle.__str__() == "0->0 1->1 2->2 3->3 4->4 5->5 6->6 7->7" for index in range(key.get_size()): assert shuffle.get_bit(key, index) == key.get_bit(index)
def test_create_shuffle_random(): Key.set_random_seed(1111) Shuffle.set_random_seed(1112) # Empty shuffle. key = Key() assert key.__str__() == "" shuffle = Shuffle(key.get_size(), Shuffle.SHUFFLE_RANDOM) assert shuffle.__str__() == "" # Non-empty shuffle. key = Key.create_random_key(16) assert key.__str__() == "0010110001010010" shuffle = Shuffle(key.get_size(), Shuffle.SHUFFLE_RANDOM) assert shuffle.__str__() == ("0->2 1->9 2->8 3->7 4->3 5->12 6->1 7->5 8->13 " "9->6 10->0 11->14 12->10 13->4 14->11 15->15") key = Key() assert key.__str__() == ""
def test_get_size(): key = Key() assert key.get_size() == 0 key = Key.create_random_key(7) assert key.get_size() == 7
def test_create_empty_key(): key = Key() assert key.get_size() == 0