def test_strata_lenth(self): """ Test the lenght of individual strata """ result = randomization.stratification([10, 12], 2) self.assertEqual(len(result[0]), 10) self.assertEqual(len(result[1]), 12)
def test_lenth(self): """ Test the overall length of the output (i.e. the number of strata) """ result = randomization.stratification([10, 10], 2) self.assertEqual(len(result), 2)