Skip to main content

Pytest - Mocking property and setter

I'm having a real hard time understanding how to mock a class that has a property and setter fixtures that access a "private" attribute.

import pytest
from pytest_mock import MockFixture
from unittest.mock import MagicMock
from typing import List

class MyEntity:

    _my_list: List[str] = []

    def __init__(self, list_vals = []):
        self._my_list = list_vals

    @property
    def my_list(self) -> List[str]:
        return self._my_list

    @my_list.setter
    def my_list(self, value: List[str]):
        self._my_list = value

    def append(self, value: str):
        self._my_list.append(value)


class Foo:

    def generate_entity(self, list_vals: List[str]) -> MyEntity:
        return MyEntity()

    def set_values(self, entity: MyEntity, list_vals: List[str]) -> MyEntity:
        entity.my_list = list_vals
        return entity

    def add_value(self, entity: MyEntity, value: str) -> MyEntity:
        entity.append(value)
        return entity


@pytest.fixture
def mock_my_entity(mocker: MockFixture) -> MagicMock:

    namespace = f"{__name__}.{MyEntity.__name__}"
    mock_my_entity = mocker.patch(namespace, autospec=True)

    return mock_my_entity.return_value

def test_foo(mock_my_entity):

    expect_list_values = ["Hello", "World"]

    foo = Foo()
    entity = foo.generate_entity(expect_list_values)
    assert len(entity._my_list) == 0

    entity = foo.set_values(entity, expect_list_values)
    assert entity._my_list == expect_list_values

    expected_extra_value = "more"
    entity = foo.add_value(entity, expected_extra_value)
    assert entity._my_list == expect_list_values + expected_extra_value

After running the test, this is what I get:

mock_my_entity = <NonCallableMagicMock name='MyEntity()' spec='MyEntity' id='4351847632'>

    def test_foo(mock_my_entity):
    
        expect_list_values = ["Hello", "World"]
    
        foo = Foo()
        entity = foo.generate_entity(expect_list_values)
        assert len(entity._my_list) == 0
    
        entity = foo.set_values(entity, expect_list_values)
>       assert entity._my_list == expect_list_values
E       AssertionError: assert <MagicMock na...='4362862544'> == ['Hello', 'World']
E         Right contains 2 more items, first extra item: 'Hello'
E         Full diff:
E         - <MagicMock name='MyEntity()._my_list' spec='list' id='4362862544'>
E         + ['Hello', 'World']

scratchpad.py:58: AssertionError

I've also tried using PropertyMock in various ways, but nothing seems to work.

How do I get this to work. How do I mock the MyEntity class so that everything works the way it should?



source https://stackoverflow.com/questions/70464598/pytest-mocking-property-and-setter

Comments

Popular posts from this blog

ValueError: X has 10 features, but LinearRegression is expecting 1 features as input

So, I am trying to predict the model but its throwing error like it has 10 features but it expacts only 1. So I am confused can anyone help me with it? more importantly its not working for me when my friend runs it. It works perfectly fine dose anyone know the reason about it? cv = KFold(n_splits = 10) all_loss = [] for i in range(9): # 1st for loop over polynomial orders poly_order = i X_train = make_polynomial(x, poly_order) loss_at_order = [] # initiate a set to collect loss for CV for train_index, test_index in cv.split(X_train): print('TRAIN:', train_index, 'TEST:', test_index) X_train_cv, X_test_cv = X_train[train_index], X_test[test_index] t_train_cv, t_test_cv = t[train_index], t[test_index] reg.fit(X_train_cv, t_train_cv) loss_at_order.append(np.mean((t_test_cv - reg.predict(X_test_cv))**2)) # collect loss at fold all_loss.append(np.mean(loss_at_order)) # collect loss at order plt.plot(np.log(al...

Sorting large arrays of big numeric stings

I was solving bigSorting() problem from hackerrank: Consider an array of numeric strings where each string is a positive number with anywhere from to digits. Sort the array's elements in non-decreasing, or ascending order of their integer values and return the sorted array. I know it works as follows: def bigSorting(unsorted): return sorted(unsorted, key=int) But I didnt guess this approach earlier. Initially I tried below: def bigSorting(unsorted): int_unsorted = [int(i) for i in unsorted] int_sorted = sorted(int_unsorted) return [str(i) for i in int_sorted] However, for some of the test cases, it was showing time limit exceeded. Why is it so? PS: I dont know exactly what those test cases were as hacker rank does not reveal all test cases. source https://stackoverflow.com/questions/73007397/sorting-large-arrays-of-big-numeric-stings

How to load Javascript with imported modules?

I am trying to import modules from tensorflowjs, and below is my code. test.html <!DOCTYPE html> <html lang="en"> <head> <meta charset="UTF-8"> <title>Document</title </head> <body> <script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@2.0.0/dist/tf.min.js"></script> <script type="module" src="./test.js"></script> </body> </html> test.js import * as tf from "./node_modules/@tensorflow/tfjs"; import {loadGraphModel} from "./node_modules/@tensorflow/tfjs-converter"; const MODEL_URL = './model.json'; const model = await loadGraphModel(MODEL_URL); const cat = document.getElementById('cat'); model.execute(tf.browser.fromPixels(cat)); Besides, I run the server using python -m http.server in my command prompt(Windows 10), and this is the error prompt in the console log of my browser: Failed to loa...