59 lines
2.3 KiB
Python
59 lines
2.3 KiB
Python
import os
|
|
import sys
|
|
import unittest
|
|
import pandas as pd
|
|
|
|
PROJECT_ROOT = os.path.realpath(os.path.join(os.path.dirname(__file__), "..", ".."))
|
|
MODULE_ROOT = os.path.join(PROJECT_ROOT, "modules")
|
|
sys.path.append(MODULE_ROOT)
|
|
from pandas_wrapper.pandas_ge import PandasGe
|
|
|
|
class TestPandasGe(unittest.TestCase):
|
|
def setUp(self):
|
|
"""Set up test data."""
|
|
self.node = PandasGe()
|
|
self.test_cases = [
|
|
# Test where all values are True (a >= b)
|
|
(
|
|
pd.DataFrame({"a": [2.0, 3.0]}, index=["row1", "row2"]),
|
|
pd.DataFrame({"a": [1.5, 2.5]}, index=["row1", "row2"]),
|
|
pd.DataFrame({"a": [True, True]}, index=["row1", "row2"])
|
|
),
|
|
# Test with equality and lt
|
|
(
|
|
pd.DataFrame({"a": [2.5, 4.5]}, index=[0, 1]),
|
|
pd.DataFrame({"a": [3.0, 4.5]}, index=[0, 1]),
|
|
pd.DataFrame({"a": [False, True]}, index=[0, 1])
|
|
),
|
|
# Test with integer column labels (True)
|
|
(
|
|
pd.DataFrame({1: [15.0, 25.0]}, index=[0, 1]),
|
|
pd.DataFrame({1: [10.0, 20.0]}, index=[0, 1]),
|
|
pd.DataFrame({1: [True, True]}, index=[0, 1])
|
|
),
|
|
# Test with a False case (a < b)
|
|
(
|
|
pd.DataFrame({"a": [7.0]}, index=["row1"]),
|
|
pd.DataFrame({"a": [7.5]}, index=["row1"]),
|
|
pd.DataFrame({"a": [False]}, index=["row1"])
|
|
),
|
|
# True case with an integer column label
|
|
(
|
|
pd.DataFrame({2: [9.5]}, index=["row1"]),
|
|
pd.DataFrame({2: [9.5]}, index=["row1"]),
|
|
pd.DataFrame({2: [True]}, index=["row1"])
|
|
),
|
|
]
|
|
|
|
def test_ge_operation(self):
|
|
"""Test greater-than-or-equal operation between two pandas DataFrames."""
|
|
for df_a, df_b, expected in self.test_cases:
|
|
with self.subTest(df_a=df_a, df_b=df_b):
|
|
retval = self.node.f(df_a, df_b)
|
|
self.assertIsInstance(retval, tuple, "Return value should be a tuple.")
|
|
self.assertEqual(len(retval), 1, "Return tuple should have a single DataFrame.")
|
|
pd.testing.assert_frame_equal(retval[0], expected)
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|