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)