Commit e9d04d06 authored by alat-rights's avatar alat-rights
Browse files

make test_bert_tokenizer use absolute path

parent a69f9428
Loading
Loading
Loading
Loading
+5 −1
Original line number Diff line number Diff line
import pytest
import os
from os.path import join


@pytest.mark.torch
@@ -35,6 +37,8 @@ def test_loading():

  loader = FASTALoader(
      featurizer=featurizer, legacy=False, auto_add_annotations=True)
  data = loader.create_dataset(input_files="./data/uniprot_truncated.fasta")
  cwd = os.getcwd()
  data = loader.create_dataset(
      input_files=join(cwd, "data/uniprot_truncated.fasta"))

  assert data.X.shape == (61, 3, 5)