From b889ddd0107ec73cd7c6d42e2997ff74dd012a48 Mon Sep 17 00:00:00 2001
From: Martin Infante <62361952+Minfante377@users.noreply.github.com>
Date: Tue, 13 Aug 2024 22:51:42 -0300
Subject: [PATCH] Run unit tests on ci (#1055)

---
 Makefile                        | 10 ++++++++--
 tests/integration/test_reads.py |  1 +
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/Makefile b/Makefile
index a16208aaca..8fe7f8a0e8 100644
--- a/Makefile
+++ b/Makefile
@@ -58,7 +58,10 @@ test-gcs:
 	sh ./dev/run-gcs-server.sh
 	poetry run  pytest tests/ -m gcs ${PYTEST_ARGS}
 
-test-coverage:
+test-coverage-unit:
+	poetry run coverage run --source=pyiceberg/ --data-file=.coverage.unit -m pytest tests/ -v -m "(unmarked or parametrize) and not integration" ${PYTEST_ARGS}
+
+test-coverage-integration:
 	docker compose -f dev/docker-compose-integration.yml kill
 	docker compose -f dev/docker-compose-integration.yml rm -f
 	docker compose -f dev/docker-compose-integration.yml up -d
@@ -67,7 +70,10 @@ test-coverage:
 	sleep 10
 	docker compose -f dev/docker-compose-integration.yml cp ./dev/provision.py spark-iceberg:/opt/spark/provision.py
 	docker compose -f dev/docker-compose-integration.yml exec -T spark-iceberg ipython ./provision.py
-	poetry run coverage run --source=pyiceberg/ -m pytest tests/ ${PYTEST_ARGS}
+	poetry run coverage run --source=pyiceberg/ --data-file=.coverage.integration -m pytest tests/ -v -m integration ${PYTEST_ARGS}
+
+test-coverage: | test-coverage-unit test-coverage-integration
+	poetry run coverage combine .coverage.unit .coverage.integration
 	poetry run coverage report -m --fail-under=90
 	poetry run coverage html
 	poetry run coverage xml
diff --git a/tests/integration/test_reads.py b/tests/integration/test_reads.py
index cbfd64e194..3cf17c0e8c 100644
--- a/tests/integration/test_reads.py
+++ b/tests/integration/test_reads.py
@@ -753,6 +753,7 @@ def test_configure_row_group_batch_size(session_catalog: Catalog) -> None:
     assert len(batches) == entries
 
 
+@pytest.mark.integration
 @pytest.mark.parametrize("catalog", [pytest.lazy_fixture("session_catalog_hive"), pytest.lazy_fixture("session_catalog")])
 def test_table_scan_default_to_large_types(catalog: Catalog) -> None:
     identifier = "default.test_table_scan_default_to_large_types"