Как исправить функцию в вызове pandas DataFrame.applyPython

Программы на Python
Ответить Пред. темаСлед. тема
Anonymous
 Как исправить функцию в вызове pandas DataFrame.apply

Сообщение Anonymous »

Я пытаюсь написать тест для класса, который выполняет применение pandas. Вот упрощенная версия:

Код: Выделить всё

class Foo:
def bar(self):
print("BAR")
return "BAR"

def apply(self, df: pd.DataFrame):
df["value"] = df.apply(self.bar, axis=1)
Вот тест, который я пытался написать:

Код: Выделить всё

import unittest
from unittest.mock import patch

from my_module import Foo

class TestFoo(unittest.TestCase):
@patch.object(Foo, "bar")
def test_apply(self, mock_bar):
df = pd.DataFrame()

mock_bar.return_value = ["value"]

foo = Foo()
updated_df = foo.apply(df)

self.assertIn("value", updated_df.columns)
mock_bar.assert_called_once()
Когда я запускаю этот тест, печатается «BAR», что указывает на то, что метод bar() не был исправлен должным образом. Что я здесь делаю не так и как это исправить?


Подробнее здесь: https://stackoverflow.com/questions/790 ... apply-call
Реклама
Ответить Пред. темаСлед. тема

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

  • Похожие темы
    Ответы
    Просмотры
    Последнее сообщение
  • В Python Pandas DataFrame.apply () метод Lambda выражается
    Anonymous » » в форуме Python
    0 Ответы
    7 Просмотры
    Последнее сообщение Anonymous
  • Как передать строку в функцию, используя df.apply в Pandas
    Anonymous » » в форуме Python
    0 Ответы
    9 Просмотры
    Последнее сообщение Anonymous
  • Как передать строку в функцию, используя df.apply в Pandas
    Anonymous » » в форуме Python
    0 Ответы
    26 Просмотры
    Последнее сообщение Anonymous
  • Как передать строку в функцию, используя df.apply в Pandas
    Anonymous » » в форуме Python
    0 Ответы
    4 Просмотры
    Последнее сообщение Anonymous
  • Как получить один Dataframe вместо нескольких Dataframe при преобразовании списка словарей в Dataframe
    Гость » » в форуме Python
    0 Ответы
    161 Просмотры
    Последнее сообщение Гость

Вернуться в «Python»