can now put file extensions or names in files data

This commit is contained in:
Silen Naihin
2023-06-25 19:30:04 -04:00
parent 2411c35d0e
commit d6a6e69f2e
6 changed files with 57 additions and 18 deletions

View File

@@ -23,10 +23,12 @@ class TestRetrieval1(RetrievalChallenge):
indirect=True,
)
def test_retrieval(self, workspace, current_challenge_data):
file = self.open_file(workspace, data.ground.files[0])
files_contents = self.open_files(workspace, data.ground.files)
score = self.scoring(file, data.ground)
scores = []
for file_content in files_contents:
score = self.scoring(file_content, data.ground)
print("Your score is:", score)
scores.append(score)
print("You score is:", score)
assert score
assert 1 in scores