Add support for OpenReview papers

1 files changed, 10 insertions, 0 deletions

diff --git a/test.py b/test.py
index 2ec59d8..201cc9f 100644
--- a/test.py
+++ b/test.py
@@ -15,6 +15,7 @@ from arxiv2remarkable import (
     ArxivProvider,
     PMCProvider,
     ACMProvider,
+    OpenReviewProvider,
     LocalFileProvider,
     PdfUrlProvider,
 )
@@ -73,6 +74,15 @@ class Tests(unittest.TestCase):
         fsize = os.path.getsize(filename)
         self.assertTrue(1691444 < fsize <= 1693444)
 
+    def test_openreview(self):
+        prov = OpenReviewProvider(upload=False)
+        url = "https://openreview.net/forum?id=S1x4ghC9tQ"
+        exp_filename = "Gregor_et_al_-_Temporal_Difference_Variational_Auto-Encoder_2018.pdf"
+        filename = prov.run(url)
+        self.assertEqual(exp_filename, os.path.basename(filename))
+        fsize = os.path.getsize(filename)
+        self.assertTrue(1110316 < fsize <= 1112316)
+
     def test_local(self):
         local_filename = "test.pdf"
         with open(local_filename, "w") as fp: