From 7e85930bf0e6dd74856d438dcb632b633af1cb83 Mon Sep 17 00:00:00 2001
From: Gertjan van den Burg <gertjanvandenburg@gmail.com>
Date: Thu, 13 Jun 2019 16:57:52 +0100
Subject: Add support for OpenReview papers

---
 test.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

(limited to 'test.py')

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:
-- 
cgit v1.2.3