59 lines
2.1 KiB
Python
59 lines
2.1 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_scalar_float import PandasGeScalarFloat
|
|
|
|
class TestPandasGeScalarFloat(unittest.TestCase):
|
|
def setUp(self):
|
|
"""Set up test data."""
|
|
self.node = PandasGeScalarFloat()
|
|
self.test_cases = [
|
|
# Test where all values are True
|
|
(
|
|
pd.DataFrame({"a": [2.5, 3.5]}, index=["row1", "row2"]),
|
|
2.1,
|
|
pd.DataFrame({"a": [True, True]}, index=["row1", "row2"])
|
|
),
|
|
# Test with mixed result
|
|
(
|
|
pd.DataFrame({"a": [3.6, 4.9]}, index=[0, 1]),
|
|
3.1,
|
|
pd.DataFrame({"a": [True, True]}, index=[0, 1])
|
|
),
|
|
# Test with integer column labels
|
|
(
|
|
pd.DataFrame({1: [14.0, 26.0]}, index=[0, 1]),
|
|
15.1,
|
|
pd.DataFrame({1: [False, True]}, index=[0, 1])
|
|
),
|
|
# Test with a False case (a < b)
|
|
(
|
|
pd.DataFrame({"a": [6.0]}, index=["row1"]),
|
|
7.1,
|
|
pd.DataFrame({"a": [False]}, index=["row1"])
|
|
),
|
|
# Another False case with an integer column label
|
|
(
|
|
pd.DataFrame({2: [7.5]}, index=["row1"]),
|
|
7.5,
|
|
pd.DataFrame({2: [True]}, index=["row1"])
|
|
),
|
|
]
|
|
|
|
def test_ge_operation(self):
|
|
"""Test greater-than 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()
|