#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Test file for: csv_to_latex.py

import os
import sys
from pathlib import Path

import pytest

# Add scripts/python to path for imports
ROOT_DIR = Path(__file__).resolve().parent.parent.parent
sys.path.insert(0, str(ROOT_DIR / "scripts" / "python"))

# Try to import pandas and the functions
try:
    import pandas as pd  # noqa: F401
    from csv_to_latex import csv_to_latex, escape_latex, format_number

    HAS_DEPS = True
except ImportError:
    HAS_DEPS = False

# Skip all tests if pandas not available
pytestmark = pytest.mark.skipif(not HAS_DEPS, reason="pandas not installed")


class TestEscapeLatex:
    """Test LaTeX special character escaping."""

    def test_escape_latex_ampersand(self):
        """Should escape ampersand."""
        result = escape_latex("A & B")
        assert result == r"A \& B"

    def test_escape_latex_percent(self):
        """Should escape percent sign."""
        result = escape_latex("50%")
        assert result == r"50\%"

    def test_escape_latex_underscore(self):
        """Should escape underscore."""
        result = escape_latex("a_b")
        assert result == r"a\_b"

    def test_escape_latex_hash(self):
        """Should escape hash/pound sign."""
        result = escape_latex("#1")
        assert result == r"\#1"

    def test_escape_latex_dollar(self):
        """Should escape dollar sign."""
        result = escape_latex("$100")
        assert result == r"\$100"

    def test_escape_latex_backslash(self):
        """Should escape backslash."""
        result = escape_latex("a\\b")
        # Braces in \textbackslash{} also get escaped
        assert result == r"a\textbackslash\{\}b"

    def test_escape_latex_braces(self):
        """Should escape curly braces."""
        result = escape_latex("a{b}c")
        assert result == r"a\{b\}c"

    def test_escape_latex_tilde(self):
        """Should escape tilde."""
        result = escape_latex("a~b")
        assert result == r"a\textasciitilde{}b"

    def test_escape_latex_caret(self):
        """Should escape caret."""
        result = escape_latex("a^b")
        assert result == r"a\textasciicircum{}b"

    def test_escape_latex_nan(self):
        """Should handle pandas NaN values."""
        result = escape_latex(float("nan"))
        assert result == ""

    def test_escape_latex_none(self):
        """Should handle None values."""
        result = escape_latex(None)
        assert result == ""

    def test_escape_latex_multiple_chars(self):
        """Should escape multiple special characters."""
        result = escape_latex("A & B: 50% #1")
        assert result == r"A \& B: 50\% \#1"

    def test_escape_latex_numeric_input(self):
        """Should convert numbers to strings and escape."""
        result = escape_latex(42)
        assert result == "42"

    def test_escape_latex_empty_string(self):
        """Should handle empty strings."""
        result = escape_latex("")
        assert result == ""


class TestFormatNumber:
    """Test number formatting."""

    def test_format_number_integer(self):
        """Should format integers without decimals."""
        result = format_number("42.0")
        assert result == "42"

    def test_format_number_decimal(self):
        """Should format decimals with up to 3 places."""
        result = format_number("3.14159")
        assert result == "3.142"

    def test_format_number_small_scientific(self):
        """Should use scientific notation for very small numbers."""
        result = format_number("0.001")
        # Should be in scientific notation
        assert "e" in result.lower()

    def test_format_number_zero(self):
        """Should format zero correctly."""
        result = format_number("0.0")
        assert result == "0"

    def test_format_number_negative(self):
        """Should handle negative numbers."""
        result = format_number("-3.14")
        assert result == "-3.14"

    def test_format_number_text(self):
        """Should return non-numeric text as-is."""
        result = format_number("hello")
        assert result == "hello"

    def test_format_number_strips_trailing_zeros(self):
        """Should strip trailing zeros."""
        result = format_number("3.100")
        assert result == "3.1"

    def test_format_number_strips_trailing_dot(self):
        """Should strip trailing decimal point."""
        result = format_number("3.000")
        assert result == "3"

    def test_format_number_large_number(self):
        """Should format large numbers correctly."""
        result = format_number("123456.789")
        assert result == "123456.789"

    def test_format_number_very_small_nonzero(self):
        """Should use scientific notation for numbers < 0.01."""
        result = format_number("0.005")
        assert "e" in result.lower()

    def test_format_number_boundary_0_01(self):
        """Should use scientific notation at 0.01 boundary."""
        result = format_number("0.009")
        assert "e" in result.lower()


class TestCsvToLatex:
    """Test full CSV to LaTeX conversion."""

    def test_csv_to_latex_creates_file(self, tmp_path):
        """Should create output LaTeX file."""
        # Create test CSV
        csv_file = tmp_path / "test.csv"
        csv_file.write_text("Name,Age\nAlice,30\nBob,25")

        output_file = tmp_path / "output.tex"

        result = csv_to_latex(csv_file, output_file)

        assert result is True
        assert output_file.exists()

    def test_csv_to_latex_contains_tabular(self, tmp_path):
        """Output should contain tabular environment."""
        csv_file = tmp_path / "test.csv"
        csv_file.write_text("A,B\n1,2")

        output_file = tmp_path / "output.tex"
        csv_to_latex(csv_file, output_file)

        content = output_file.read_text()
        assert r"\begin{tabular}" in content
        assert r"\end{tabular}" in content

    def test_csv_to_latex_contains_headers(self, tmp_path):
        """Output should contain bold headers."""
        csv_file = tmp_path / "test.csv"
        csv_file.write_text("Name,Age\nAlice,30")

        output_file = tmp_path / "output.tex"
        csv_to_latex(csv_file, output_file)

        content = output_file.read_text()
        # Headers should be bold and title-cased
        assert r"\textbf{Name}" in content
        assert r"\textbf{Age}" in content

    def test_csv_to_latex_contains_data(self, tmp_path):
        """Output should contain table data."""
        csv_file = tmp_path / "test.csv"
        csv_file.write_text("Name,Value\nTest,42")

        output_file = tmp_path / "output.tex"
        csv_to_latex(csv_file, output_file)

        content = output_file.read_text()
        assert "Test" in content
        assert "42" in content

    def test_csv_to_latex_escapes_special_chars(self, tmp_path):
        """Should escape LaTeX special characters in data."""
        csv_file = tmp_path / "test.csv"
        csv_file.write_text("Text\nA & B")

        output_file = tmp_path / "output.tex"
        csv_to_latex(csv_file, output_file)

        content = output_file.read_text()
        # Ampersand should be escaped
        assert r"\&" in content

    def test_csv_to_latex_contains_table_env(self, tmp_path):
        """Output should contain table environment."""
        csv_file = tmp_path / "test.csv"
        csv_file.write_text("A\n1")

        output_file = tmp_path / "output.tex"
        csv_to_latex(csv_file, output_file)

        content = output_file.read_text()
        assert r"\begin{table}" in content
        assert r"\end{table}" in content

    def test_csv_to_latex_contains_caption(self, tmp_path):
        """Output should contain caption."""
        csv_file = tmp_path / "test.csv"
        csv_file.write_text("A\n1")

        output_file = tmp_path / "output.tex"
        csv_to_latex(csv_file, output_file)

        content = output_file.read_text()
        assert r"\caption{" in content

    def test_csv_to_latex_custom_caption(self, tmp_path):
        """Should use custom caption if provided."""
        csv_file = tmp_path / "test.csv"
        csv_file.write_text("A\n1")

        output_file = tmp_path / "output.tex"
        custom_caption = "My Custom Caption"
        csv_to_latex(csv_file, output_file, caption=custom_caption)

        content = output_file.read_text()
        assert custom_caption in content

    def test_csv_to_latex_custom_label(self, tmp_path):
        """Should use custom label if provided."""
        csv_file = tmp_path / "test.csv"
        csv_file.write_text("A\n1")

        output_file = tmp_path / "output.tex"
        custom_label = "tab:mylabel"
        csv_to_latex(csv_file, output_file, label=custom_label)

        content = output_file.read_text()
        assert r"\label{tab:mylabel}" in content

    def test_csv_to_latex_invalid_csv(self, tmp_path):
        """Should return False for invalid CSV."""
        csv_file = tmp_path / "nonexistent.csv"
        output_file = tmp_path / "output.tex"

        result = csv_to_latex(csv_file, output_file)

        assert result is False
        assert not output_file.exists()

    def test_csv_to_latex_contains_booktabs(self, tmp_path):
        """Should use booktabs rules (toprule, midrule, bottomrule)."""
        csv_file = tmp_path / "test.csv"
        csv_file.write_text("A,B\n1,2")

        output_file = tmp_path / "output.tex"
        csv_to_latex(csv_file, output_file)

        content = output_file.read_text()
        assert r"\toprule" in content
        assert r"\midrule" in content
        assert r"\bottomrule" in content

    def test_csv_to_latex_number_formatting(self, tmp_path):
        """Should format numbers appropriately."""
        csv_file = tmp_path / "test.csv"
        csv_file.write_text("Value\n3.14159\n42.0")

        output_file = tmp_path / "output.tex"
        csv_to_latex(csv_file, output_file)

        content = output_file.read_text()
        # Float should be formatted
        assert "3.142" in content
        # Integer should not have decimals
        assert "42 " in content or "42\\\\" in content

    def test_csv_to_latex_column_alignment(self, tmp_path):
        """Should align numeric columns right."""
        csv_file = tmp_path / "test.csv"
        csv_file.write_text("Name,Age\nAlice,30")

        output_file = tmp_path / "output.tex"
        csv_to_latex(csv_file, output_file)

        content = output_file.read_text()
        # Should have tabular with alignment spec (l for text, r for numbers)
        assert r"\begin{tabular}{" in content

    def test_csv_to_latex_handles_missing_values(self, tmp_path):
        """Should handle missing/NaN values."""
        csv_file = tmp_path / "test.csv"
        csv_file.write_text("A,B\n1,\n,3")

        output_file = tmp_path / "output.tex"
        csv_to_latex(csv_file, output_file)

        content = output_file.read_text()
        # Missing values should be represented (as --)
        assert "--" in content

    def test_csv_to_latex_truncation_message(self, tmp_path):
        """Should add truncation message for large tables."""
        # Create CSV with many rows
        lines = ["Value"] + [str(i) for i in range(50)]
        csv_file = tmp_path / "test.csv"
        csv_file.write_text("\n".join(lines))

        output_file = tmp_path / "output.tex"
        csv_to_latex(csv_file, output_file, max_rows=10)

        content = output_file.read_text()
        # Should mention truncation
        assert "truncated" in content.lower() or "omitted" in content.lower()


if __name__ == "__main__":
    pytest.main([os.path.abspath(__file__), "-v"])