[92b703]: / tests / test_dataset_assumptions.py

Download this file

19 lines (14 with data), 544 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
import pytest
import os
import pandas as pd
metadata = pd.read_csv("metadata.csv")
def test_no_duplicate_images():
filenames = metadata["filename"]
assert len(set(filenames)) == len(filenames)
def test_no_unlinked_images():
linked_images = metadata["filename"]
all_images = os.listdir("images")
#The only unlinked image should be the warning image
assert set(all_images).difference(set(linked_images)) == set(['FAFA-A1BF-49A8-A1D3-66FAFA41B7345D.jpg'])
def test_no_pediatric():
assert min(metadata["age"]) >= 18