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)
