def _create_dataset(): examples_file, label_file = BytesIO(), BytesIO() examples_file.name = 'examples.csv' label_file.name = 'labels.csv' iris = load_iris() numpy.savetxt(examples_file, iris.data, delimiter=',') numpy.savetxt(label_file, iris.target, delimiter=',') examples_file.seek(0), label_file.seek(0) return examples_file, label_file
def test_dataset_post_dataset_length_mismatch(self): examples_file, label_file = BytesIO(), BytesIO() examples_file.name = 'examples.csv' label_file.name = 'labels.csv' iris = load_iris() breast_cancer = load_breast_cancer() numpy.savetxt(examples_file, iris.data, delimiter=',') numpy.savetxt(label_file, breast_cancer.target, delimiter=',') examples_file.seek(0), label_file.seek(0) client = DjangoClient() response = client.post(reverse('datasets'), data={ 'dataset': 'TEST', 'file[0]': examples_file, 'file[1]': label_file }) self.assertEqual(400, response.status_code) self.assertEqual(b'"Examples and labels are not the same length"', response.content)