当前位置：文江博客话题详情

在Python中获得一个功能

发布于 2025-02-10 21:06:53 字数 290 浏览 2 评论 0原文

我的功能可以从数据框架中获取列计数。

函数运行后，我想确保列计数与输入列计数相同。

因此，我在功能，colinput和coloutput中有两个变量。

当我在新文件中运行UNITSEST文件时，我正在插入函数（让我们称其为“ getWeight”（），

因此，从mainscript impert get

Weepaight中，我有两个列计数计算得出的两个列计数，

我如何编写一个比较colinput ==的测试的测试。？

coloutput

原文

分享到QQ

分享到微博

如果你对这篇内容有疑问，欢迎到本站社区发帖提问参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。

发布评论

需要登录才能够评论，你可以免费注册一个本站的账号。

同展鸳鸯锦 2025-02-17 21:06:53

将正在测试的计算分解为纯函数。确保MAIN可以导入而无需副作用（使用'__ -Main __'警卫）：

main.py：

import pandas as pd

def getweight(x: pd.DataFrame) -> pd.DataFrame:
  # ...

if __name__ == '__main__':
  # run your computations

导入您感兴趣的主模块和测试属性：

unittest.py：

import pandas as pd
import main
import pytest

@pytest.mark.parametrize(
  'x',
  [
    # specify interesting inputs ...
  ]
)
def test_getweight_preserves_column_count(x: pd.DataFrame):
  assert len(x.columns) == len(main.getweight(x).columns)

*未测试

Factor out the computation under test into a pure function. Make sure main is importable without side effects (use '__main__' guards):

main.py:

import pandas as pd

def getweight(x: pd.DataFrame) -> pd.DataFrame:
  # ...

if __name__ == '__main__':
  # run your computations

Import main module and test properties you are interested in:

unittest.py:

import pandas as pd
import main
import pytest

@pytest.mark.parametrize(
  'x',
  [
    # specify interesting inputs ...
  ]
)
def test_getweight_preserves_column_count(x: pd.DataFrame):
  assert len(x.columns) == len(main.getweight(x).columns)

* not tested

回复收藏 0 原文

~没有更多了~