import os

import matplotlib.pyplot as plt


def test_example():
    plt.ion()
    cwd = os.getcwd()
    os.chdir("examples")
    exec(open("example.py").read())
    plt.close('all')
    os.chdir(cwd)
    return


def test_example_menyanthes():
    plt.ion()
    cwd = os.getcwd()
    os.chdir("examples")
    exec(open("example_menyanthes.py").read())
    plt.close('all')
    os.chdir(cwd)
    return


def test_example_project():
    plt.ion()
    cwd = os.getcwd()
    os.chdir("examples")
    exec(open("example_project.py").read())
    plt.close('all')
    os.chdir(cwd)
    return


def test_example_docs():
    plt.ion()
    cwd = os.getcwd()
    os.chdir("examples")
    exec(open("example_docs.py").read())
    plt.close('all')
    os.chdir(cwd)
    return


def test_example_stats():
    plt.ion()
    cwd = os.getcwd()
    os.chdir("examples")
    exec(open("example_stats.py").read())
    plt.close('all')
    os.chdir(cwd)
    return


def test_example_no_conv():
    plt.ion()
    cwd = os.getcwd()
    os.chdir("examples")
    exec(open("example_no_conv.py").read())
    plt.close('all')
    os.chdir(cwd)
    return


def test_example_WellModel():
    plt.ion()
    cwd = os.getcwd()
    os.chdir("examples")
    exec(open("example_WellModel.py").read())
    plt.close('all')
    os.chdir(cwd)
    return


def test_example_timestep_weighted_resample():
    plt.ion()
    cwd = os.getcwd()
    os.chdir("examples")
    exec(open("example_timestep_weighted_resample.py").read())
    plt.close('all')
    return os.chdir(cwd)

@@ -934,33 +934,78 @@
 wd)%0A    return%0A%0A
-%0A
+# TODO Make faster.Too slow for testing now%0A# 
 def test_example
@@ -1012,24 +1012,26 @@
 _no_conv():%0A
+# 
     plt.ion(
@@ -1024,32 +1024,34 @@
 #     plt.ion()%0A
+# 
     cwd = os.get
@@ -1048,32 +1048,34 @@
 d = os.getcwd()%0A
+# 
     os.chdir(%22ex
@@ -1075,36 +1075,38 @@
 dir(%22examples%22)%0A
+#
     
+ 
 exec(open(%22examp
@@ -1121,32 +1121,34 @@
 nv.py%22).read())%0A
+# 
     plt.close('a
@@ -1144,36 +1144,38 @@
 lt.close('all')%0A
+#
     
+ 
 os.chdir(cwd)%0A  
@@ -1164,32 +1164,34 @@
   os.chdir(cwd)%0A
+# 
     return%0A%0A%0Adef

remove no_conv from travis test, too slow