2.0 first commit

test/test_returns.py

@@ -0,0 +1,41 @@
+import unittest
+import pandas as pd
+from lightweight_charts import Chart
+import asyncio
+
+from util import BARS, Tester
+
+
+
+class TestReturns(Tester):
+    def test_screenshot_returns_value(self):
+        self.chart.set(BARS)
+        self.chart.show()
+        screenshot_data = self.chart.screenshot()
+        self.assertIsNotNone(screenshot_data)
+
+    def test_save_drawings(self):
+        self.chart.exit()
+
+        async def main():
+            asyncio.create_task(self.chart.show_async());
+
+            await asyncio.sleep(2)
+            self.chart.toolbox.drawings.clear() # clear drawings in python
+            self.assertTrue(len(self.chart.toolbox.drawings) == 0)
+            self.chart.run_script(f'{self.chart.id}.toolBox.saveDrawings();')
+            await asyncio.sleep(1)  # resave them, and assert they exist
+            self.assertTrue(len(self.chart.toolbox.drawings) > 0)
+            self.chart.exit()
+
+        self.chart = Chart(toolbox=True, debug=True)
+        self.chart.set(BARS)
+        self.chart.topbar.textbox('symbol', 'SYM', align='right')
+        self.chart.toolbox.save_drawings_under(self.chart.topbar['symbol'])
+        self.chart.toolbox.import_drawings("drawings.json")
+        self.chart.toolbox.load_drawings("SYM")
+        asyncio.run(main())
+
+
+if __name__ == '__main__':
+    unittest.main()