def test_permute_rows2():
h = Hub([0, 1, 2, 3, 4], [0, 1, 2, [3, 4]], [1, 2, 3, 4, 5], 0)
h.connection_table = np.array([[1, 1, 0, 0, 0], [1, 1, 0, 0, 0],
[0, 0, 1, 1, 1], [0, 1, 1, 1, 0],
[0, 0, 1, 1, 1]])
h.permute_rows()
assert h.flat_sources == [0, 1, 3, 4, 2]
assert h.perm == [0, 1, 3, 4, 2]
# noinspection PyTypeChecker
assert np.all(h.connection_table == np.array(
[[1, 1, 0, 0, 0], [1, 1, 0, 0, 0], [0, 1, 1, 1, 0], [0, 0, 1, 1, 1],
[0, 0, 1, 1, 1]]))
def test_permute_rows2():
h = Hub([0, 1, 2, 3, 4], [0, 1, 2, [3, 4]], [1, 2, 3, 4, 5], 0)
h.connection_table = np.array([
[1, 1, 0, 0, 0],
[1, 1, 0, 0, 0],
[0, 0, 1, 1, 1],
[0, 1, 1, 1, 0],
[0, 0, 1, 1, 1]])
h.permute_rows()
assert h.flat_sources == [0, 1, 3, 4, 2]
assert h.perm == [0, 1, 3, 4, 2]
# noinspection PyTypeChecker
assert np.all(h.connection_table == np.array([
[1, 1, 0, 0, 0],
[1, 1, 0, 0, 0],
[0, 1, 1, 1, 0],
[0, 0, 1, 1, 1],
[0, 0, 1, 1, 1]]))
