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)