def test_bad_arraystack(): t_array_list, expected_shape = make_testing_data.makeRandomArrays() t_array_list.append(numpy.zeros((1, 1, 1))) with pytest.raises(ValueError): riomucho.utils.array_stack(t_array_list)
def test_arraystack(): t_array_list, expected_shape = make_testing_data.makeRandomArrays() stacked = riomucho.utils.array_stack(t_array_list) assert stacked.shape == expected_shape