def test_n_treatments(self):
""" Test the number of treatments per subject
"""
results = randomization.random_treatment_order(100, 2)
for result in results:
self.assertEqual(len(result), 2)
def test_lenth(self):
""" Test the overall length of the output
"""
result = randomization.random_treatment_order(100, 2)
self.assertEqual(len(result), 100)
