From 244d0f51eb01086ff008c355cff8dba32eb58843 Mon Sep 17 00:00:00 2001
From: Gertjan van den Burg <gertjanvandenburg@gmail.com>
Date: Tue, 27 Oct 2020 22:50:35 +0100
Subject: Add provider for Nature

---
 tests/test_providers.py | 15 +++++++++++++++
 tests/test_ui.py        |  6 ++++++
 2 files changed, 21 insertions(+)

(limited to 'tests')

diff --git a/tests/test_providers.py b/tests/test_providers.py
index e701234..def77d0 100644
--- a/tests/test_providers.py
+++ b/tests/test_providers.py
@@ -20,6 +20,7 @@ from paper2remarkable.providers import (
     HTML,
     JMLR,
     LocalFile,
+    Nature,
     NBER,
     NeurIPS,
     OpenReview,
@@ -355,6 +356,20 @@ class TestProviders(unittest.TestCase):
         filename = prov.run(url)
         self.assertEqual(exp, os.path.basename(filename))
 
+    def test_nature_1(self):
+        prov = Nature(upload=False, verbose=VERBOSE)
+        url = "https://www.nature.com/articles/s41598-020-75456-0"
+        exp = "Golozar_et_al_-_Direct_Observation_of_Lithium_Metal_Dendrites_With_Ceramic_Solid_Electrolyte_2020.pdf"
+        filename = prov.run(url)
+        self.assertEqual(exp, os.path.basename(filename))
+
+    def test_nature_2(self):
+        prov = Nature(upload=False, verbose=VERBOSE)
+        url = "https://www.nature.com/articles/s41599-019-0371-1.pdf"
+        exp = "Leroi_et_al_-_On_Revolutions_2020.pdf"
+        filename = prov.run(url)
+        self.assertEqual(exp, os.path.basename(filename))
+
 
 if __name__ == "__main__":
     unittest.main()
diff --git a/tests/test_ui.py b/tests/test_ui.py
index 835f594..a1eb372 100644
--- a/tests/test_ui.py
+++ b/tests/test_ui.py
@@ -24,6 +24,7 @@ from paper2remarkable.providers import (
     HTML,
     JMLR,
     LocalFile,
+    Nature,
     NBER,
     NeurIPS,
     OpenReview,
@@ -180,6 +181,11 @@ class TestUI(unittest.TestCase):
                 "https://openaccess.thecvf.com/content_cvpr_2018/html/Cheng_Dual_Skipping_Networks_CVPR_2018_paper.html",
                 "https://openaccess.thecvf.com/content_cvpr_2018/html/Cheng_Dual_Skipping_Networks_CVPR_2018_paper.html",
             ),
+            (
+                Nature,
+                "https://www.nature.com/articles/s41599-019-0349-z",
+                "https://www.nature.com/articles/s41599-019-0349-z",
+            ),
         ]
         for exp_prov, url, exp_url in tests:
             prov, new_url, jar = choose_provider(url)
-- 
cgit v1.2.3