def test_get_component_slice_pair_3(self):
# Components reversed, no total comps
comp_slice, num_comps = get_component_slice(pair_comps[::-1])
self.assertTrue(
comp_slice == slice(pair_comps[1], pair_comps[0], None))
self.assertTrue(num_comps == pair_comps[1] - pair_comps[0])
def test_get_component_slice_pair_2(self):
# Components, total comps
comp_slice, num_comps = get_component_slice(pair_comps, total_comps)
self.assertTrue(
comp_slice == slice(pair_comps[0], pair_comps[1], None))
self.assertTrue(num_comps == pair_comps[1] - pair_comps[0])
def test_get_component_slice_int_3(self):
# Component outside total comps
comp_slice, num_comps = get_component_slice(total_comps + 5,
total_comps)
self.assertTrue(comp_slice == slice(0, total_comps, None))
self.assertTrue(num_comps == total_comps)
def test_get_component_slice_slice_1(self):
# Slice, no total comps
comp_slice, num_comps = get_component_slice(slice_comps)
self.assertTrue(comp_slice == slice_comps)
self.assertTrue(num_comps == np.arange(slice_comps.stop +
1)[slice_comps].size)
def test_get_component_slice_tuple_2(self):
# Tuple of integers, total comps
comp_slice, num_comps = get_component_slice(tuple(array_comps),
total_comps)
self.assertTrue(comp_slice == array_comps.tolist())
self.assertTrue(num_comps == array_comps.size)
def test_get_component_slice_list_1(self):
# List of integers, no total comps
comp_slice, num_comps = get_component_slice(array_comps.tolist())
self.assertTrue(comp_slice == array_comps.tolist())
self.assertTrue(num_comps == array_comps.size)
def test_get_component_slice_float(self):
# Float component, should error
with self.assertRaises(TypeError):
comp_slice, num_comps = get_component_slice(float(int_comps), total_comps)
def test_get_component_slice_int_2(self):
# Component with total comps
comp_slice, num_comps = get_component_slice(int_comps, total_comps)
self.assertTrue(comp_slice == slice(0, int_comps, None))
self.assertTrue(num_comps == int_comps)
