diff options
Diffstat (limited to 'test.py')
| -rw-r--r-- | test.py | 10 |
1 files changed, 10 insertions, 0 deletions
@@ -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(2349734 < fsize <= 2351734) + 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: |
