예제 #1
0
def all_defined_1():
    #                 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
    barr = flex.int([1, 1, 1, 0, 1, 1, 0, 0, 1, 0, 0, 0, 1, 1, 1, 1]).as_bool()
    iarr = [
        flex.size_t([0]),
        flex.size_t([1, 2]),
        flex.size_t([4, 5]),
        flex.size_t([8]),
        flex.size_t([12, 13, 14]),
        flex.size_t([15])
    ]
    oarr = [[flex.size_t([0]), flex.size_t([1, 2])], [flex.size_t([4, 5])],
            [flex.size_t([8])], [flex.size_t([12, 13, 14]),
                                 flex.size_t([15])]]
    return refinement_flags.manager(individual_sites=True,
                                    torsion_angles=True,
                                    rigid_body=True,
                                    individual_adp=True,
                                    group_adp=True,
                                    tls=True,
                                    occupancies=True,
                                    group_anomalous=True,
                                    sites_individual=barr,
                                    sites_torsion_angles=barr,
                                    sites_rigid_body=iarr,
                                    adp_individual_iso=barr,
                                    adp_individual_aniso=barr,
                                    adp_group=iarr,
                                    group_h=iarr,
                                    adp_tls=iarr,
                                    s_occupancies=oarr)
예제 #2
0
def exercise_deepcopy_show_select():
    sel_all = flex.int([1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]).as_bool()
    sel_none = flex.int([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]).as_bool()
    sel_all_true = flex.int([0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 0, 0]).as_bool()
    sel_all_false = flex.int([1, 0, 0, 1, 0, 0, 1, 0, 0, 1, 0, 0, 1, 0, 1, 1]).as_bool()
    sel_mix = flex.int([1, 1, 0, 0, 1, 1, 0, 0, 0, 1, 1, 0, 0, 0, 1, 1]).as_bool()
    #
    compare(rm=all_defined(), expected_result=expected_result_all)
    compare(rm=all_defined(), expected_result=expected_result_all, deep_copy=True)
    compare(rm=all_defined(), expected_result=expected_result_all, selection=sel_all)
    compare(rm=all_defined(), expected_result=expected_result_none_true, selection=sel_none)
    compare(rm=all_defined(), expected_result=expected_result_all, selection=sel_all_true)
    compare(rm=all_defined(), expected_result=expected_result_none_true, selection=sel_all_false)
    compare(rm=all_defined(), expected_result=expected_result_mix, selection=sel_mix)
    #
    compare(rm=refinement_flags.manager(), expected_result=expected_result_none_false)
    compare(rm=refinement_flags.manager(), expected_result=expected_result_none_false, deep_copy=True)
    compare(rm=refinement_flags.manager(), expected_result=expected_result_none_false, selection=sel_all)
    compare(rm=refinement_flags.manager(), expected_result=expected_result_none_false, selection=sel_none)
    compare(rm=refinement_flags.manager(), expected_result=expected_result_none_false, selection=sel_all_true)
    compare(rm=refinement_flags.manager(), expected_result=expected_result_none_false, selection=sel_all_false)
    compare(rm=refinement_flags.manager(), expected_result=expected_result_none_false, selection=sel_mix)
    #
    rm = all_defined()
    rm_dc = rm.deep_copy()
    rm_dc.individual_sites = False
    rm_dc.torsion_angles = False
    rm_dc.rigid_body = False
    rm_dc.individual_adp = False
    rm_dc.group_adp = False
    rm_dc.tls = False
    rm_dc.occupancies = False
    rm_dc.group_anomalous = False
    rm_dc.sites_individual = None
    rm_dc.sites_torsion_angles = None
    rm_dc.sites_rigid_body = None
    rm_dc.adp_individual_iso = None
    rm_dc.adp_individual_aniso = None
    rm_dc.adp_group = None
    rm_dc.group_h = None
    rm_dc.adp_tls = None
    rm_dc.s_occupancies = None
    out = StringIO()
    rm.show(log=out)
    assert out.getvalue() == expected_result_all
    out = StringIO()
    rm_dc.show(log=out)
    assert out.getvalue() == expected_result_none_false
예제 #3
0
def all_defined_1():
    #                 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
    barr = flex.int([1, 1, 1, 0, 1, 1, 0, 0, 1, 0, 0, 0, 1, 1, 1, 1]).as_bool()
    iarr = [
        flex.size_t([0]),
        flex.size_t([1, 2]),
        flex.size_t([4, 5]),
        flex.size_t([8]),
        flex.size_t([12, 13, 14]),
        flex.size_t([15]),
    ]
    oarr = [
        [flex.size_t([0]), flex.size_t([1, 2])],
        [flex.size_t([4, 5])],
        [flex.size_t([8])],
        [flex.size_t([12, 13, 14]), flex.size_t([15])],
    ]
    return refinement_flags.manager(
        individual_sites=True,
        torsion_angles=True,
        rigid_body=True,
        individual_adp=True,
        group_adp=True,
        tls=True,
        occupancies=True,
        group_anomalous=True,
        sites_individual=barr,
        sites_torsion_angles=barr,
        sites_rigid_body=iarr,
        adp_individual_iso=barr,
        adp_individual_aniso=barr,
        adp_group=iarr,
        group_h=iarr,
        adp_tls=iarr,
        s_occupancies=oarr,
    )
예제 #4
0
def exercise_deepcopy_show_select():
    sel_all = flex.int([1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
                        1]).as_bool()
    sel_none = flex.int([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
                         0]).as_bool()
    sel_all_true = flex.int([0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 1, 0, 1, 0,
                             0]).as_bool()
    sel_all_false = flex.int([1, 0, 0, 1, 0, 0, 1, 0, 0, 1, 0, 0, 1, 0, 1,
                              1]).as_bool()
    sel_mix = flex.int([1, 1, 0, 0, 1, 1, 0, 0, 0, 1, 1, 0, 0, 0, 1,
                        1]).as_bool()
    #
    compare(rm=all_defined(), expected_result=expected_result_all)
    compare(rm=all_defined(),
            expected_result=expected_result_all,
            deep_copy=True)
    compare(rm=all_defined(),
            expected_result=expected_result_all,
            selection=sel_all)
    compare(rm=all_defined(),
            expected_result=expected_result_none_true,
            selection=sel_none)
    compare(rm=all_defined(),
            expected_result=expected_result_all,
            selection=sel_all_true)
    compare(rm=all_defined(),
            expected_result=expected_result_none_true,
            selection=sel_all_false)
    compare(rm=all_defined(),
            expected_result=expected_result_mix,
            selection=sel_mix)
    #
    compare(rm=refinement_flags.manager(),
            expected_result=expected_result_none_false)
    compare(rm=refinement_flags.manager(),
            expected_result=expected_result_none_false,
            deep_copy=True)
    compare(rm=refinement_flags.manager(),
            expected_result=expected_result_none_false,
            selection=sel_all)
    compare(rm=refinement_flags.manager(),
            expected_result=expected_result_none_false,
            selection=sel_none)
    compare(rm=refinement_flags.manager(),
            expected_result=expected_result_none_false,
            selection=sel_all_true)
    compare(rm=refinement_flags.manager(),
            expected_result=expected_result_none_false,
            selection=sel_all_false)
    compare(rm=refinement_flags.manager(),
            expected_result=expected_result_none_false,
            selection=sel_mix)
    #
    rm = all_defined()
    rm_dc = rm.deep_copy()
    rm_dc.individual_sites = False
    rm_dc.torsion_angles = False
    rm_dc.rigid_body = False
    rm_dc.individual_adp = False
    rm_dc.group_adp = False
    rm_dc.tls = False
    rm_dc.occupancies = False
    rm_dc.group_anomalous = False
    rm_dc.sites_individual = None
    rm_dc.sites_torsion_angles = None
    rm_dc.sites_rigid_body = None
    rm_dc.adp_individual_iso = None
    rm_dc.adp_individual_aniso = None
    rm_dc.adp_group = None
    rm_dc.group_h = None
    rm_dc.adp_tls = None
    rm_dc.s_occupancies = None
    out = StringIO()
    rm.show(log=out)
    assert out.getvalue() == expected_result_all
    out = StringIO()
    rm_dc.show(log=out)
    assert out.getvalue() == expected_result_none_false