kindle manager

test_kman.py

@@ -116,7 +116,6 @@ class TestKman(unittest.TestCase):
         self.assertEqual(t_ds, '2020/1/13 20:11:05')
 
     # test function format_data
-    def test_format_data(self):
     def test_format_data(self):
         t_books = self.cre_tbooks()
         t_out = format_data(t_books, ft='MD')
@@ -125,6 +124,11 @@ class TestKman(unittest.TestCase):
         self.assertEqual(t_out[2], 'HL|薛兆丰经济学讲义 |薛兆丰|2020/1/13 8:11:05|边际就是“新增”带来的“新增”。\n')
         t_out.clear()
 
+    def test_add_note_to_highlight(self):
+        t_books = self.cre_tbooks()
+        t_books_remove_nt = add_note_to_highlight(t_books)
+        for k in t_books_remove_nt.keys():
+            bn = k
         self.assertEqual((t_books_remove_nt[bn]['1']['content']).replace('\n',''),\
                 '边际就是“新增”带来的“新增”。'+NTPREF+ '山寨 假货 问题')