def test_checkattr(simple_ring): assert checkattr('Length')(simple_ring[0]) is True assert checkattr('not_an_attr')(simple_ring[0]) is False assert (list(filter(checkattr('Length', 1), simple_ring)) == [simple_ring[0], simple_ring[3], simple_ring[4], simple_ring[5]]) assert list(filter(checkattr('Length', 2), simple_ring)) == [] assert list(filter(checkattr('not_an_attr'), simple_ring)) == []
def test_get_cells(simple_ring): a = numpy.ones(6, dtype=bool) numpy.testing.assert_equal(get_cells(simple_ring, checkattr('Length')), a) a = numpy.array([False, True, False, False, False, False]) numpy.testing.assert_equal(get_cells(simple_ring, 'attr'), a) a = numpy.array([True, False, False, False, False, False]) numpy.testing.assert_equal(get_cells(simple_ring, 'FamName', 'D1'), a)