diff --git a/examples/data.pickle b/examples/data.pickle deleted file mode 100644 index 4c73e94..0000000 Binary files a/examples/data.pickle and /dev/null differ diff --git a/test_convoys.py b/test_convoys.py index bce3688..c52e8ba 100644 --- a/test_convoys.py +++ b/test_convoys.py @@ -208,21 +208,3 @@ def test_marriage_example(): def test_dob_violations_example(): from examples.dob_violations import run run() - - -def test_other_data(): - data = pandas.read_pickle('examples/data.pickle') - unit, groups, (G, B, T) = convoys.utils.get_arrays( - data, groups='group', created='created_at', converted='action_at', - now='now', unit='days', max_groups=10, group_min_size=100) - convoys.plotting.plot_cohorts(G, B, T, - model='generalized-gamma', - groups=groups, - ci=0.95) - matplotlib.pyplot.legend() - convoys.plotting.plot_cohorts(G, B, T, - model='kaplan-meier', - groups=groups, - plot_kwargs={'linestyle': '--'}) - matplotlib.pyplot.xlabel(unit) - matplotlib.pyplot.savefig('test.png')