def test_notebook(self):
results = list(extractor.Extractor(self.ipynb_storage))
assert len(results) == 2
assert results[0][0] == 3141
assert results[1][0] == 2
solution = extractor.Extractor(self.pyfiles_storage[1]).file_content
solution = solution.replace('# Upload 3141', '')
assert results[0][1][0].code == solution.strip()
def test_image(self):
results = list(extractor.Extractor(self.image_storage))
assert len(results) == 1
assert results[0][0] == 3
with pytest.raises(BadUploadFile) as e_info:
list(extractor.Extractor(self.image_no_exercise_storage))
assert e_info.type is BadUploadFile
assert e_info.value.args[0] == "Can't resolve exercise id."
def test_py(self):
for file in self.pyfiles_storage:
solutions = list(extractor.Extractor(file))
assert len(solutions) == 1
assert solutions[0][0] == 3141
with pytest.raises(BadUploadFile) as e_info:
list(extractor.Extractor(self.pyfile_no_exercise_storage))
assert e_info.type is BadUploadFile
assert e_info.value.args[0] == "Can't resolve exercise id."
def test_py(self):
for file in self.pyfiles_storage:
solutions = list(extractor.Extractor(file))
assert len(solutions) == 1
assert solutions[0][0] == 3141
