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)