test_files.py

import pytest
from api import files


def test_extension():
    assert files.guess_type_from_filename('example.pdf') == 'pdf'

def test_multi_extension():
    assert files.guess_type_from_filename('example.zip') == 'archive'
    assert files.guess_type_from_filename('example.gephysio.zip') == 'gephysio'

def test_nifti():
    assert files.guess_type_from_filename('example.nii') == 'nifti'
    assert files.guess_type_from_filename('example.nii.gz') == 'nifti'
    assert files.guess_type_from_filename('example.nii.x.gz') == None

def test_qa():
    assert files.guess_type_from_filename('example.png') == 'image'
    assert files.guess_type_from_filename('example.qa.png') == 'qa'
    assert files.guess_type_from_filename('example.qa') == None
    assert files.guess_type_from_filename('example.qa.png.unknown') == None

def test_tabular_data():
    assert files.guess_type_from_filename('example.csv') == 'tabular data'
    assert files.guess_type_from_filename('example.csv.gz') == 'tabular data'
    assert files.guess_type_from_filename('example.tsv') == 'tabular data'
    assert files.guess_type_from_filename('example.tsv.gz') == 'tabular data'

def test_unknown():
    assert files.guess_type_from_filename('example.unknown') == None