def test_get_valid_one_arg(self): df = Study(1).sample_template.to_dataframe() df = df[['ph', 'country']] df = {idx: [row['country']] for idx, row in df.iterrows()} exp = {'header': ['country'], 'samples': df} response = self.get('/api/v1/study/1/samples/categories=country', headers=self.headers) self.assertEqual(response.code, 200) obs = json_decode(response.body) self.assertEqual(obs, exp)