[CI] Add tests.py

.travis.yml

@@ -31,6 +31,6 @@ install:
 
 # command to run tests
 script:
-  - for VERSION in $PYTHON_VERSIONS ; do source activate $VERSION ; done
+  - for VERSION in $PYTHON_VERSIONS ; do source activate $VERSION ; py.test tests.py ; done
 
 sudo: false

README.md

@@ -92,7 +92,7 @@ A `.report.txt` file will also be generated that contains a summary of
 number of people affected by over-expanded TREDs as well as population allele
 frequency.
 
-To better understand the uncertainties in the prediction, one call plot the
+To better understand the uncertainties in the prediction, we can plot the
 likelihood surface based on the model. Using the same example as above at the
 Huntington disease case, we can run a command on the JSON output, with option
 `--tred HD` to specify the locus.

tests.py

@@ -0,0 +1,23 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+
+
+def test_tred():
+    """ Run tred.py on sample CSV file and generate TSV file with the genotype
+    """
+    from tredparse.tred import main
+    main(["tests/samples.csv", "--workdir", "work"])
+
+
+def test_tredreport():
+    """ Highlight the potential risk individuals
+    """
+    from tredparse.tredreport import main
+    main(["work/t001.json", "work/t002.json", "--tsv", "work.tsv"])
+
+
+def test_tredplot():
+    """ Plot the likelihood surface based on the model
+    """
+    from tredparse.tredplot import likelihood
+    likelihood(["work/t001.json", "--tred", "HD"])