def test_square(): op = Operator(shapein=3, shapeout=3) assert op.flags.square op._reset(shapeout=4) assert not op.flags.square op = Operator(shapein=3, shapeout=3) op._reset(shapein=4) assert not op.flags.square