update unittest to work with operations

GjjvdBurg · Nov 14, 2024 · 3332f86 · 3332f86
1 parent 98c7c8a
commit 3332f86
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/paper2remarkable/providers/arxiv.py b/paper2remarkable/providers/arxiv.py
@@ -40,7 +40,9 @@ def __init__(self, *args, **kwargs):
         self.informer = ArxivInformer()
 
         # register the dearxiv operation
-        self.operations.insert(0, ("dearxiv", self.dearxiv))
+        for format in self.operations:
+            if format in "pdf ps".split():
+                self.operations[format].insert(0, ("dearxiv", self.dearxiv))
 
     def get_abs_pdf_urls(self, url):
         """Get the pdf and abs url from any given arXiv url"""