Merge pull request #66 from DanielGoldfarb/master

check inputs are float, and avoid "Image" confusion (IPython.display vs. PIL).

Author: DanielGoldfarb

examples/savefig.ipynb

@@ -77,7 +77,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from IPython.display import Image"
+    "import IPython.display as IPydisplay"
    ]
   },
   {
@@ -89,7 +89,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "-rw-rw-rw- 1 dino dino 14697 Mar  3 21:17 \u001b[0m\u001b[01;35mtestsave.png\u001b[0m\r\n"
+      "-rw-rw-rw- 1 dino dino 14697 Mar 23 20:02 \u001b[0m\u001b[01;35mtestsave.png\u001b[0m\r\n"
      ]
     },
     {
@@ -106,7 +106,7 @@
    ],
    "source": [
     "%ls -l testsave.png\n",
-    "Image(filename='testsave.png')"
+    "IPydisplay.Image(filename='testsave.png')"
    ]
   },
   {
@@ -157,7 +157,7 @@
     }
    ],
    "source": [
-    "Image(buf.read())"
+    "IPydisplay.Image(buf.read())"
    ]
   },
   {
@@ -198,8 +198,8 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "-rw-rw-rw- 1 dino dino 12287 Mar  3 21:17 \u001b[0m\u001b[01;35mtsave30.jpg\u001b[0m\n",
-      "-rw-rw-rw- 1 dino dino 59840 Mar  3 21:17 \u001b[0m\u001b[01;35mtsave100.jpg\u001b[0m\n"
+      "-rw-rw-rw- 1 dino dino 12287 Mar 23 20:02 \u001b[0m\u001b[01;35mtsave30.jpg\u001b[0m\n",
+      "-rw-rw-rw- 1 dino dino 59840 Mar 23 20:02 \u001b[0m\u001b[01;35mtsave100.jpg\u001b[0m\n"
      ]
     },
     {
@@ -228,8 +228,8 @@
    "source": [
     "%ls -l tsave30.jpg\n",
     "%ls -l tsave100.jpg\n",
-    "Image(filename='tsave30.jpg')\n",
-    "Image(filename='tsave100.jpg')"
+    "IPydisplay.Image(filename='tsave30.jpg')\n",
+    "IPydisplay.Image(filename='tsave100.jpg')"
    ]
   },
   {
@@ -279,7 +279,7 @@
    ],
    "source": [
     "_ = buf30dpi.seek(0)\n",
-    "Image(buf30dpi.read())"
+    "IPydisplay.Image(buf30dpi.read())"
    ]
   },
   {
@@ -301,7 +301,7 @@
    ],
    "source": [
     "_ = buf100dpi.seek(0)\n",
-    "Image(buf100dpi.read())"
+    "IPydisplay.Image(buf100dpi.read())"
    ]
   },
   {

src/mplfinance/_arg_validators.py

@@ -25,16 +25,23 @@ def _check_and_prepare_data(data):
     if not isinstance(data.index,pd.core.indexes.datetimes.DatetimeIndex):
         raise TypeError('Expect data.index as DatetimeIndex')
 
+    cols = ['Open','High','Low','Close']
+
     dates   = mdates.date2num(data.index.to_pydatetime())
     opens   = data['Open'].values
     highs   = data['High'].values
     lows    = data['Low'].values
     closes  = data['Close'].values
     if 'Volume' in data.columns:
         volumes = data['Volume'].values
+        cols.append('Volume')
     else:
         volumes = None
 
+    for col in cols:
+        if not all( isinstance(v,(float,int)) for v in data[col] ):
+            raise ValueError('Data for column "'+str(col)+'" must be ALL float or int.')
+
     return dates, opens, highs, lows, closes, volumes