#!/bin/sh
set -eu
test_dir=$(mktemp -d)
trap 'rm -rf "$test_dir"' EXIT HUP INT TERM
cd "$test_dir"
mhctools --help
mhctools --mhc-predictor random --mhc-alleles HLA-A0201 \
    --sequence SIINFEKLQY --extract-subsequences --mhc-peptide-lengths 9 \
    --output-csv predictions.csv
printf '>example\nSIINFEKLQY\n' > peptides.fasta
mhctools --mhc-predictor random --mhc-alleles HLA-A0201 \
    --input-fasta-file peptides.fasta --mhc-peptide-lengths 9 \
    --output-csv fasta-predictions.csv
python3 - <<'PYTEST'
import pandas as pd
for filename in ('predictions.csv', 'fasta-predictions.csv'):
    frame = pd.read_csv(filename)
    assert len(frame) == 2, frame
    assert set(frame.peptide) == {'SIINFEKLQ', 'IINFEKLQY'}, frame
    assert set(frame.allele) == {'HLA-A*02:01'}, frame
    assert frame.affinity.between(0, 50000).all(), frame
PYTEST
