Python mocking can't get right return_value

Question

I have the following function with image a python object from a module and having the method shape.

def foo(image):
    print image.shape

I want to mock the object image and it's return value

image = Mock()
image.shape.return_value = (0,0)
foo(image)
>>> <Mock name='mock.shape' id='4416382544'>

Calling foo(image), image.shape will return . How can I make it return (0,0)? (I need to get the value to test some condition like if image.shape = (0,0). I'm struggling to have it work.

Thanks for your help

Also, shape isn't being accessed as a function -- It's a property. You should be able to just do image.shape = (0, 0) where image is the Mock instance. — mgilson
– mgilson, Commented Jul 23, 2014 at 6:04

user2357112 · Accepted Answer · 2014-07-23 06:04:00Z

1

image.shape = (0, 0)

Set the attribute directly. return_value has nothing to do with attribute accesses.

answered Jul 23, 2014 at 6:04

user2357112

286k32 gold badges490 silver badges571 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Romanzo Criminale Over a year ago

Thanks. Even if i'm using def foo(image): return image.shape ? I'm not sure to see what return_value is for then

user2357112 Over a year ago

@RomanzoCriminale: return_value is for when you're calling something like a function. image.shape.return_value would be what you'd get if you tried image.shape() rather than image.shape.

Collectives™ on Stack Overflow

Python mocking can't get right return_value

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related