Fix get_bytes and add a test. Closes #3

Author: typpo

quickchart/__init__.py

@@ -66,7 +66,7 @@ def get_short_url(self):
 
     def get_bytes(self):
         resp = self._post('https://quickchart.io/chart')
-        return response.content
+        return resp.content
 
     def to_file(self, path):
         content = self.get_bytes()

tests.py

@@ -33,5 +33,21 @@ def test_no_chart(self):
 
         self.assertRaises(RuntimeError, qc.get_url)
 
+    def test_get_bytes(self):
+        qc = QuickChart()
+        qc.width = 600
+        qc.height = 300
+        qc.config = {
+            "type": "bar",
+            "data": {
+                "labels": ["Hello world", "Test"],
+                "datasets": [{
+                    "label": "Foo",
+                    "data": [1, 2]
+                }]
+            }
+        }
+        self.assertTrue(len(qc.get_bytes()) > 8000)
+
 if __name__ == '__main__':
     unittest.main()