def test_convert_bool_list(): bool_list = util.convert_bool_list(n=10, k=None) assert len(bool_list) == 10 assert bool_list[0] == True bool_list = util.convert_bool_list(n=10, k=3) assert len(bool_list) == 10 assert bool_list[3] is True assert bool_list[2] is False bool_list = util.convert_bool_list(n=10, k=[3, 7]) assert len(bool_list) == 10 assert bool_list[3] is True assert bool_list[7] is True assert bool_list[2] is False bool_list = util.convert_bool_list(n=3, k=[False, False, True]) assert len(bool_list) == 3 assert bool_list[0] is False assert bool_list[1] is False assert bool_list[2] is True bool_list = util.convert_bool_list(n=3, k=[]) assert len(bool_list) == 3 assert bool_list[0] is False
def _bool_list(self, k=None): """ returns a bool list of the length of the lens models if k = None: returns bool list with True's if k is int, returns bool list with False's but k'th is True if k is a list of int, e.g. [0, 3, 5], returns a bool list with True's in the integers listed and False elsewhere if k is a boolean list, checks for size to match the numbers of models and returns it :param k: None, int, or list of ints :return: bool list """ return convert_bool_list(n=self._num_func, k=k)
def test_raise(self): with self.assertRaises(ValueError): array = np.ones(5) util.array2image(array) with self.assertRaises(ValueError): array = np.ones((2, 2)) util.array2cube(array, 2, 2) with self.assertRaises(ValueError): x, y = np.ones(6), np.ones(6) util.get_axes(x, y) with self.assertRaises(ValueError): util.selectBest(array=np.ones(6), criteria=np.ones(5), numSelect=1, highest=True) with self.assertRaises(ValueError): util.select_best(array=np.ones(6), criteria=np.ones(5), num_select=1, highest=True) with self.assertRaises(ValueError): util.convert_bool_list(n=2, k=[3, 7]) with self.assertRaises(ValueError): util.convert_bool_list(n=3, k=[True, True]) with self.assertRaises(ValueError): util.convert_bool_list(n=2, k=[0.1, True])