SS 2025

2025_Research_Software_Engineering_2/tests/test_utils.py

@@ -0,0 +1,94 @@
+"""Unit tests for utils.py"""
+
+from unittest.mock import Mock
+import pytest
+import pandas as pd
+import geopandas as gpd
+
+import utils
+
+
+def test_is_number():
+    """Test if strings are correctly identified as numbers or not."""
+    assert utils.is_number("42") is True
+    assert utils.is_number("3.14") is True
+    assert utils.is_number("abc") is False
+    assert utils.is_number("") is False
+
+
+def test_parse_number():
+    """Test parsing numeric strings to int or float and raising errors for invalid input."""
+    assert utils.parse_number("42") == 42
+    assert utils.parse_number("3.14") == 3.14
+    with pytest.raises(ValueError):
+        utils.parse_number("not a number")
+
+
+def test_request_response():
+    """Test HTTP GET request and check if response has expected structure."""
+    response = utils.request_response("https://httpbin.org/get")
+    assert response.status_code == 200
+    assert "url" in response.json()
+
+
+def test_parse_html():
+    """Test parsing HTML content and extracting elements."""
+    mock_response = Mock()
+    mock_response.text = "<html><body><p>Hello</p></body></html>"
+    result = utils.parse_html(mock_response)
+    assert result.p.text == "Hello"
+
+
+def test_read_file(local_tmp_path):
+    """Test reading content from a file."""
+    test_path = local_tmp_path / "test.txt"
+    test_path.write_text("test content")
+    content = utils.read_file(test_path)
+    assert content == "test content"
+
+
+def test_save_file(local_tmp_path):
+    """Test saving content to a file."""
+    test_path = local_tmp_path / "output.txt"
+    utils.save_file("saved content", test_path)
+    assert test_path.read_text() == "saved content"
+
+
+def test_read_df(local_tmp_path):
+    """Test reading a CSV file into a pandas DataFrame."""
+    csv_path = local_tmp_path / "test.csv"
+    csv_path.write_text("a,b\n1,2\n3,4")
+    df = utils.read_df(csv_path, seperator=",", header_line=0, icol=None)
+    assert not df.empty
+
+
+def test_save_df(local_tmp_path):
+    """Test saving a pandas DataFrame to a CSV file."""
+    df = pd.DataFrame({"x": [1, 2], "y": [3, 4]})
+    path = local_tmp_path / "output.csv"
+    utils.save_df(df, path)
+    loaded = pd.read_csv(path, index_col=0)
+    pd.testing.assert_frame_equal(df, loaded)
+
+
+def test_read_gdf(local_tmp_path):
+    """Test reading a GeoDataFrame from a GeoPackage file."""
+    gdf = gpd.GeoDataFrame(
+        {'col': [1]}, geometry=gpd.points_from_xy([0], [0]), crs="EPSG:4326")
+    path = local_tmp_path / "test.gpkg"
+    gdf.to_file(path, driver="GPKG")
+
+    result = utils.read_gdf(path)
+    assert isinstance(result, gpd.GeoDataFrame)
+    assert result.equals(gdf)
+
+
+def test_save_gdf(local_tmp_path):
+    """Test saving a GeoDataFrame to a GeoPackage file."""
+    gdf = gpd.GeoDataFrame(
+        {'col': [1]}, geometry=gpd.points_from_xy([0], [0]), crs="EPSG:4326")
+    path = local_tmp_path / "output.gpkg"
+    utils.save_gdf(gdf, path)
+    loaded = gpd.read_file(path)
+    assert isinstance(loaded, gpd.GeoDataFrame)
+    assert loaded.equals(gdf)