@with_default_shape
    def eye_mask(self, shape):
        return ~np.eye(*shape, dtype=bool)
Build a mask using np.eye.
dataset/ETHPy150Open quantopian/zipline/tests/pipeline/base.py/BasePipelineTestCase.eye_mask