add continuous integration, fix missing files, add badges to main pages (#2)

* add appveyor configuration

* fix missing dependencies for pytest

* add coremltools

* travis + fix coremltools (windows)

* update dependencies

* add scipy

* fix dependencies, add missing files

* update ci scripts

* fix appveyor, travis, add circleci

* add badge, fix unit test scripts

* disable circlecli

* add coverage (token should be changed)

* update appveyor command line

* remove unneeded file, update command line appveyor

* replace one assert by an exception, update command lines

* update travis appveyor buid

* fix appveyor python path

* uplaod coverage report, update readme with badges

* update test data

* update onnx model

* update unit test file

* update readme

* update readme

Author: xadupre

.appveyor.yml

@@ -0,0 +1,42 @@
+environment:
+
+  global:
+    # SDK v7.0 MSVC Express 2008's SetEnv.cmd script will fail if the
+    # /E:ON and /V:ON options are not enabled in the batch script intepreter
+    # See: http://stackoverflow.com/a/13751649/163740
+    WITH_COMPILER: "cmd /E:ON /V:ON /C .\\appveyor\\run_with_compiler.cmd"
+
+  matrix:
+
+    - PYTHON: "C:\\Python36-x64"
+      PYTHON_VERSION: "3.6.x"
+      PYTHON_ARCH: "64"
+
+init:
+    - "ECHO %PYTHON% %PYTHON_VERSION% %PYTHON_ARCH%"
+
+install:
+    - "%PYTHON%\\python -m pip install --upgrade pip"
+    - "%PYTHON%\\Scripts\\pip install -r requirements.txt"
+    - "%PYTHON%\\Scripts\\pip install git+https://github.com/apple/coremltools"
+    - "%PYTHON%\\Scripts\\pip install https://github.com/xadupre/xgboost/releases/download/0.7/xgboost-0.7-cp36-cp36m-win_amd64.whl"
+    - "%PYTHON%\\Scripts\\pip install -r requirements-dev.txt"
+    - set PYTHONPATH=%~dp0
+
+build: off
+
+test_script:
+    - "%PYTHON%\\python -m pytest --cov=onnxmltools --cov-report=term --cov-report=html --cov-report=xml tests"
+
+after_test:
+    - "%PYTHON%\\python -u setup.py bdist_wheel"
+    - if not exist dist mkdir dist
+    - if not exist dist\htmlcov mkdir dist\htmlcov
+    - xcopy htmlcov dist\htmlcov
+    - copy *.xml dist\htmlcov
+    - if %APPVEYOR_ACCOUNT_NAME%==xadupre %PYTHON%\Scripts\codecov -f coverage.xml -t 2c916354-0f91-46b4-a577-f80b0031c941
+
+artifacts:
+    - path: dist
+      name: onnxmltools
+    

.coveragerc

@@ -0,0 +1,2 @@
+[run]
+omit = tests/*

.gitignore

@@ -0,0 +1,31 @@
+# Visual Studio Code files
+.vscode
+
+# IPython notebook checkpoints
+.ipynb_checkpoints
+
+# Compiled python
+*.pyc
+
+# setup.py intermediates
+.eggs
+dist
+
+# PyCharm files
+.idea
+
+# OSX dir files
+.DS_Store
+
+# Windows
+*.bat
+
+# test generated files
+.pytest_cache
+.cache
+tests/utils/models/coreml_OneHotEncoder_BikeSharing_new.json
+tests/utils/models/coreml_OneHotEncoder_BikeSharing2.onnx
+tests/baseline/outmodels
+htmlcov
+coverage.xml
+.coverage

.travis.yml

@@ -0,0 +1,12 @@
+dist: trusty
+language: python
+python:
+    - "3.5"
+    - "3.6"
+install: 
+    - pip install -r requirements.txt
+    - pip install -r requirements-dev.txt
+    - export PYTHONPATH=pwd
+script:
+    - python -m pytest --cov=onnxmltools --cov-report=term --cov-report=html --cov-report=xml tests
+

README.md

@@ -1,5 +1,10 @@
 <p align="center"><img width="40%" src="docs/ONNXMLTools_logo_main.png" /></p>
 
+| Linux | Windows |
+|-------|---------|
+| [![Build Status](https://travis-ci.org/onnx/onnxmltools.svg?branch=master)](https://travis-ci.org/onnx/onnxmltools) | [![Build status](https://ci.appveyor.com/api/projects/status/d1xav3amubypje4n?svg=true)](https://ci.appveyor.com/project/xadupre/onnxmltools) |
+
+
 # Introduction 
 ONNXMLTools enables you to convert models from different machine learning toolkits into [ONNX](https://onnx.ai). Currently the following toolkits are supported:
 * Apple CoreML

README.rst

@@ -1,6 +1,17 @@
 
 .. image:: https://github.com/onnx/onnxmltools/blob/master/docs/ONNXMLTools_logo_main.png
-    
+
+.. list-table::
+   :widths: 4 4
+   :header-rows: 0
+   
+   * - .. image:: https://travis-ci.org/onnx/onnxmltools.svg?branch=master
+            :target: https://travis-ci.org/onnx/onnxmltools
+            :alt: Build Status Linux
+     - .. image:: https://ci.appveyor.com/api/projects/status/d1xav3amubypje4n?svg=true
+            :target: https://ci.appveyor.com/project/xadupre/onnxmltools
+            :alt: Build Status Windows
+
 Introduction 
 ============
 
@@ -31,7 +42,6 @@ Example
 
 Here is a simple example to convert a CoreML model:
 
-
 ::
 
     import onnxmltools

onnxmltools/.gitignore

@@ -84,7 +84,7 @@ def convert(context, cm_node, inputs, outputs, prefix):
         nb.add_attribute('base_values', cm_node.treeEnsembleRegressor.treeEnsemble.basePredictionValue)
         post_transform = get_onnx_tree_post_transform(cm_node.treeEnsembleRegressor.postEvaluationTransform)
     else:
-        assert (False, "Unknown tree type!")
+        raise TypeError("Unknown tree type: prefix='{0}'".format(prefix))
 
     leaf_treeids = [node.treeId for node in nodes if 6 == node.nodeBehavior for weight in node.evaluationInfo]
     leaf_nodeids = [node.nodeId for node in nodes if 6 == node.nodeBehavior for weight in node.evaluationInfo]

onnxmltools/convert/coreml/TreeConverterCommon.py

@@ -0,0 +1,11 @@
+codecov
+coremltools
+pandas
+pytest
+pytest-cov
+scikit-learn
+scipy
+wheel
+xgboost
+
+