def test_format_table_returns_correct_dimensions_single_dim_present(self):
variable_idxs = [2] # order of table
num_states = (3, ) # current shape of table
correct_states = (1, 1, 3) # correct shape of table
table = np.random.rand(*num_states)
new_table = utils.format_table(variable_idxs, table)
assert correct_states == new_table.shape, 'Returned table is not correct shape'
def test_format_table_returns_correct_dimensions_all_dims_present_2(self):
variable_idxs = [0, 2, 1] # order of table
num_states = (5, 2, 3) # shape of table
correct_states = (5, 3, 2)
table = np.random.rand(*num_states)
new_table = utils.format_table(variable_idxs, table)
assert correct_states == new_table.shape, 'Returned table is not correct shape'
def test_format_table_returns_correct_dimensions_when_not_all_dims_present(
self):
variable_idxs = [3, 0]
num_states = (5, 2)
correct_states = (2, 1, 1, 5)
table = np.random.rand(*num_states)
new_table = utils.format_table(variable_idxs, table)
assert correct_states == new_table.shape, 'Returned table is not correct shape'
def test_reorder_array_returns_correctly_reordered_2(self):
test = np.arange(81).reshape((3, 3, 3, 3))
swaps = [(0, 1), (3, 0), (1, 2)]
order = [3, 2, 0, 1]
swapped = test.copy()
for swap in swaps:
swapped = swapped.swapaxes(swap[0], swap[1])
reordered = utils.format_table([3, 2, 0, 1], swapped)
assert np.all(np.isclose(test, reordered))
def test_format_table_returns_correct_dimensions_when_var_set_supplied(
self):
variable_idxs = [3, 0]
num_states = (5, 2)
var_set = set(range(6))
correct_states = (2, 1, 1, 5, 1, 1)
table = np.random.rand(*num_states)
new_table = utils.format_table(variable_idxs, table, var_set=var_set)
assert correct_states == new_table.shape, 'Returned table is not correct shape'
def test_reorder_array_returns_correctly_reordered(self):
test = np.arange(8).reshape((2, 2, 2))
swapped = test.swapaxes(2, 1)
reordered = utils.format_table([0, 2, 1], swapped)
assert np.all(np.isclose(test, reordered))
def test_format_table_returns_copy_of_original(self):
table = np.array([[1, 2], [3, 4]])
new_table = utils.format_table([0, 1], table)
assert table is not new_table