From 3247e5b4f6a1486a84591f4a8d672a63747cab82 Mon Sep 17 00:00:00 2001 From: lucas-nelson-uiuc Date: Fri, 27 Dec 2024 13:44:24 -0600 Subject: [PATCH] tests: include pandas, pyarrow to testing workflow --- .github/workflows/pytest.yml | 2 +- tests/core/test_filters.py | 16 ++++++++-------- 2 files changed, 9 insertions(+), 9 deletions(-) diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml index cf5742b..019b1b6 100644 --- a/.github/workflows/pytest.yml +++ b/.github/workflows/pytest.yml @@ -22,7 +22,7 @@ jobs: run: | python -m pip install --upgrade pip pip install pyspark==${{ matrix.pyspark-version }} attrs==${{ matrix.attrs-version }} - pip install --upgrade pytest loguru pandas + pip install --upgrade pytest loguru pandas pyarrow pip install --editable . - name: Run testing suite run: python -m pytest tests diff --git a/tests/core/test_filters.py b/tests/core/test_filters.py index 398f9a4..d50ba66 100644 --- a/tests/core/test_filters.py +++ b/tests/core/test_filters.py @@ -59,8 +59,8 @@ def test_filter_regex(self, eits_data): ), ) assertDataFrameEqual( - ~( - eits_data.filter( + eits_data.filter( + ~( F.col("title").rlike(TEST_PATTERN) | F.col("comments").rlike(TEST_PATTERN) ) @@ -70,8 +70,8 @@ def test_filter_regex(self, eits_data): ), ) assertDataFrameEqual( - ~( - eits_data.filter( + eits_data.filter( + ~( F.col("title").rlike(TEST_PATTERN) & F.col("comments").rlike(TEST_PATTERN) ) @@ -113,8 +113,8 @@ def test_filter_elements(self, eits_data): ), ) assertDataFrameEqual( - ~( - eits_data.filter( + eits_data.filter( + ~( F.col("formats").isin(TEST_ELEMENTS) | F.col("producer").isin(TEST_ELEMENTS) ) @@ -124,8 +124,8 @@ def test_filter_elements(self, eits_data): ), ) assertDataFrameEqual( - ~( - eits_data.filter( + eits_data.filter( + ~( F.col("formats").isin(TEST_ELEMENTS) & F.col("producer").isin(TEST_ELEMENTS) )