From 8c7efef6d1bdc0842cd080c96f42a82a4ed057a3 Mon Sep 17 00:00:00 2001 From: Gabriel Selzer Date: Mon, 24 Jan 2022 15:29:05 -0600 Subject: [PATCH] Use pytest.importorskip for pandas --- tests/test_return_widgets.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/tests/test_return_widgets.py b/tests/test_return_widgets.py index cbf5ddf51..e82e3bca1 100644 --- a/tests/test_return_widgets.py +++ b/tests/test_return_widgets.py @@ -7,13 +7,12 @@ from inspect import signature import numpy as np -import pandas as pd import pytest from magicgui import magicgui, widgets -def _dataframe_equals(object1: pd.DataFrame, object2: pd.DataFrame): +def _dataframe_equals(object1, object2): assert object1.equals(object2) @@ -25,10 +24,17 @@ def _default_equals(object1, object2): assert object1 == object2 +def _generate_pandas_test_data(): + pd = pytest.importorskip( + "pandas", reason="Pandas required for a return widget test" + ) + return pd.DataFrame({"Res1": [1, 2, 3], "Res2": [4, 5, 6]}) + + parameterizations = [ # pandas dataframe ( - pd.DataFrame({"Res1": [1, 2, 3], "Res2": [4, 5, 6]}), + _generate_pandas_test_data(), widgets.Table, _dataframe_equals, ),