classroom ========= search_first_allpassrow ----------------------- >>> from classroom import search_first_allpassrow >>> gD = { ... 10: ['JBB', 0.0], ... 15: ['GWV', 5.5], ... 22: ['JJJ', 6.5], ... 27: ['AWV', 3.5], ... 31: ['MMP', 5.5], ... 35: ['BWV', 4.5], ... 42: ['SCC', 7.5], ... 49: ['BWV', 9.0], ... 55: ['FCC', 5.0], ... 72: ['ABC', 7.0], ... 81: ['GMM', 8.0], ... 92: ['MRP', 10.0], ... 99: ['ARG', 10.0], ... } >>> cl1 = [ ... [22, -1, 42, 81, -1, 10, -1], ... [15, -1, 27, -1, -1, 35, 49], ... [92, -1, 72, -1, 31, 55, 99], ... ] >>> sol = search_first_allpassrow(gD, cl1) >>> sol # doctesttag: +TAG=1_classroom-1_search_first_allpassrow (3, [92, -1, 72, -1, 31, 55, 99]) >>> cl2 = [ ... [22, -1, 42, 81, -1, 10, -1], ... [15, -1, 72, -1, -1, 35, 49], ... [92, -1, 27, -1, 31, 55, 99], ... ] >>> sol = search_first_allpassrow(gD, cl2) >>> sol # doctesttag: +TAG=1_classroom-1_search_first_allpassrow (-1, []) >>> cl3 = [ ... [-1, -1, -1, -1, -1, -1, -1], ... [-1, -1, -1, -1, -1, -1, -1], ... [-1, -1, -1, -1, -1, -1, -1], ... ] >>> sol = search_first_allpassrow(gD, cl3) >>> sol # doctesttag: +TAG=1_classroom-1_search_first_allpassrow (-1, []) >>> cl4 = [ ... [22, -1, 42, 81, -1, 10, -1], ... [15, -1, 22, -1, -1, 31, 49], ... [92, -1, 72, -1, 31, 55, 99], ... ] >>> sol = search_first_allpassrow(gD, cl4) >>> sol # doctesttag: +TAG=1_classroom-1_search_first_allpassrow (2, [15, -1, 22, -1, -1, 31, 49]) search_first_allpassrow_plus ---------------------------- >>> from classroom import search_first_allpassrow_plus >>> sol = search_first_allpassrow_plus(gD, cl1) >>> sol # doctesttag: +TAG=1_classroom-2_search_first_allpassrow_plus (3, [10.0, 7.0, 5.5, 5.0, 10.0]) >>> sol = search_first_allpassrow_plus(gD, cl2) >>> sol # doctesttag: +TAG=1_classroom-2_search_first_allpassrow_plus (-1, []) >>> sol = search_first_allpassrow_plus(gD, cl3) >>> sol # doctesttag: +TAG=1_classroom-2_search_first_allpassrow_plus (-1, []) >>> sol = search_first_allpassrow_plus(gD, cl4) >>> sol # doctesttag: +TAG=1_classroom-2_search_first_allpassrow_plus (2, [5.5, 6.5, 5.5, 9.0]) transit ======= >>> from transit import gen_transitD >>> trajD = { ... 'H11':[1, 5, 7, 9, 6, 2, 6, 2, 6, 2, 6, 2], ... 'M22':[3, 1, 5, 7, 9, 6], ... 'H23':[5, 7, 9, 6, 2, 9, 6], ... } >>> transitD = gen_transitD(trajD) >>> corrD = { ... (1, 5): ['H11', 'M22'], ... (5, 7): ['H11', 'H23', 'M22'], ... (7, 9): ['H11', 'H23', 'M22'], ... (9, 6): ['H11', 'H23', 'H23', 'M22'], ... (6, 2): ['H11', 'H11', 'H11', 'H11', 'H23'], ... (2, 6): ['H11', 'H11', 'H11'], ... (3, 1): ['M22'], ... (2, 9): ['H23'], ... } >>> transitD == corrD # doctesttag: +TAG=2_transit-1_gen_transitD True >>> trajD = { ... 'H55': [1, 3, 5, 9, 5, 1], ... 'M22': [3, 5, 9, 5, 9, 5, 1], ... 'H13': [1, 3, 1, 3, 5, 9, 5, 1], ... 'A10': [1, 3, 1, 3, 5, 9, 5, 1] ... } >>> transitD = gen_transitD(trajD) >>> corrD = { ... (1, 3): ['A10', 'A10', 'H13', 'H13', 'H55'], ... (3, 5): ['A10', 'H13', 'H55', 'M22'], ... (5, 9): ['A10', 'H13', 'H55', 'M22', 'M22'], ... (9, 5): ['A10', 'H13', 'H55', 'M22', 'M22'], ... (5, 1): ['A10', 'H13', 'H55', 'M22'], ... (3, 1): ['A10', 'H13'], ... } >>> transitD == corrD # doctesttag: +TAG=2_transit-1_gen_transitD True >>> trajD = { ... 'H11':[1, 5, 1, 5, 1, 5], ... } >>> transitD = gen_transitD(trajD) >>> corrD = { ... (1, 5): ['H11', 'H11', 'H11'], ... (5, 1): ['H11', 'H11'], ... } >>> transitD == corrD # doctesttag: +TAG=2_transit-1_gen_transitD True >>> trajD = {} >>> transitD = gen_transitD(trajD) >>> transitD == {} # doctesttag: +TAG=2_transit-1_gen_transitD True sort_transitD ------------- >>> from transit import sort_transitD >>> transitD = { ... (1, 5): ['H11', 'M22'], ... (5, 7): ['H11', 'H23', 'M22'], ... (7, 9): ['H11', 'H23', 'M22'], ... (9, 6): ['H11', 'H23', 'H23', 'M22'], ... (6, 2): ['H11', 'H11', 'H11', 'H11', 'H23'], ... (2, 6): ['H11', 'H11', 'H11'], ... (3, 1): ['M22'], ... (2, 9): ['H23'], ... } >>> L = sort_transitD(transitD) >>> L # doctesttag: +TAG=2_transit-2_sort_transitD [((6, 2), ['H11', 'H11', 'H11', 'H11', 'H23']), ((9, 6), ['H11', 'H23', 'H23', 'M22']), ((2, 6), ['H11', 'H11', 'H11']), ((5, 7), ['H11', 'H23', 'M22']), ((7, 9), ['H11', 'H23', 'M22']), ((1, 5), ['H11', 'M22']), ((2, 9), ['H23']), ((3, 1), ['M22'])] >>> transitD = { ... (1, 3): ['A10', 'A10', 'H13', 'H13', 'H55'], ... (3, 5): ['A10', 'H13', 'H55', 'M22'], ... (5, 9): ['A10', 'H13', 'H55', 'M22', 'M22'], ... (9, 5): ['A10', 'H13', 'H55', 'M22', 'M22'], ... (5, 1): ['A10', 'H13', 'H55', 'M22'], ... (3, 1): ['A10', 'H13'], ... } >>> L = sort_transitD(transitD) >>> L # doctesttag: +TAG=2_transit-2_sort_transitD [((1, 3), ['A10', 'A10', 'H13', 'H13', 'H55']), ((5, 9), ['A10', 'H13', 'H55', 'M22', 'M22']), ((9, 5), ['A10', 'H13', 'H55', 'M22', 'M22']), ((3, 5), ['A10', 'H13', 'H55', 'M22']), ((5, 1), ['A10', 'H13', 'H55', 'M22']), ((3, 1), ['A10', 'H13'])] >>> transitD = { ... (2, 5): ['H11', 'H11', 'H11'], ... (2, 3): ['H11', 'H11', 'H11'], ... (1, 3): ['H11', 'H11', 'H11'], ... (1, 1): ['H11', 'H11', 'H11'], ... (1, 2): ['H11', 'H11', 'H11'], ... (1, 0): ['H11', 'H11', 'H11', 'H11'], ... } >>> L = sort_transitD(transitD) >>> L # doctesttag: +TAG=2_transit-2_sort_transitD [((1, 0), ['H11', 'H11', 'H11', 'H11']), ((1, 1), ['H11', 'H11', 'H11']), ((1, 2), ['H11', 'H11', 'H11']), ((1, 3), ['H11', 'H11', 'H11']), ((2, 3), ['H11', 'H11', 'H11']), ((2, 5), ['H11', 'H11', 'H11'])] filter_pat_temp =============== >>> from pat_temp import filter_pat_temp >>> n = filter_pat_temp('patients.csv', 'patients-380.csv', 38.0) >>> with open('patients-380.ref', 'r') as fref, open('patients-380.csv', 'r') as f: ... l = f.readline() ... lref = fref.readline() ... while l != '' and lref == l: ... l = f.readline() ... lref = fref.readline() ... oeq = lref == l == '' >>> n == 8 and oeq # doctesttag: +TAG=3_filter_pat_temp True >>> n = filter_pat_temp('patients.csv', 'patients-390.csv', 39.0) >>> with open('patients-390.ref', 'r') as fref, open('patients-390.csv', 'r') as f: ... l = f.readline() ... lref = fref.readline() ... while l != '' and lref == l: ... l = f.readline() ... lref = fref.readline() ... oeq = lref == l == '' >>> n == 4 and oeq # doctesttag: +TAG=3_filter_pat_temp True >>> n = filter_pat_temp('patients.csv', 'patients-393.csv', 39.3) >>> with open('patients-393.ref', 'r') as fref, open('patients-393.csv', 'r') as f: ... l = f.readline() ... lref = fref.readline() ... while l != '' and lref == l: ... l = f.readline() ... lref = fref.readline() ... oeq = lref == l == '' >>> n == 2 and oeq # doctesttag: +TAG=3_filter_pat_temp True >>> n = filter_pat_temp('patients.csv', 'patients-395.csv', 39.5) >>> with open('patients-395.ref', 'r') as fref, open('patients-395.csv', 'r') as f: ... l = f.readline() ... lref = fref.readline() ... while l != '' and lref == l: ... l = f.readline() ... lref = fref.readline() ... oeq = lref == l == '' >>> n == 0 and oeq # doctesttag: +TAG=3_filter_pat_temp True recassim ======== >>> from recassim import recassim >>> s = recassim(0, 1) >>> s # doctesttag: +TAG=4_recassim 1 >>> s = recassim(0, 7) >>> s # doctesttag: +TAG=4_recassim 5 >>> s = recassim(0, 20) >>> s # doctesttag: +TAG=4_recassim 7 >>> s = recassim(0, 0) >>> s # doctesttag: +TAG=4_recassim 0 >>> s = recassim(0, 11) >>> s # doctesttag: +TAG=4_recassim 10 >>> s = recassim(0, 42) >>> s # doctesttag: +TAG=4_recassim 20 >>> s = recassim(7, 11) >>> s # doctesttag: +TAG=4_recassim 20 >>> s = recassim(10, 1) >>> s # doctesttag: +TAG=4_recassim 9 >>> s = recassim(10, 200) >>> s # doctesttag: +TAG=4_recassim 724