[SELF-23] Brazil tariff engine tests

{
  "cost": {
    "input_tokens": 176,
    "estimated_usd": 0.095598,
    "output_tokens": 6338
  },
  "model": "claude-sonnet-4-20250514",
  "status": "success",
  "transport": "anthropic_api",
  "latency_ms": 80448,
  "response_text": "I'll create comprehensive unit tests for the Brazil tariff engine. Let me first examine the source file to understand the implementation, then create appropriate tests.\n\n<file_path>c8_platform/tests/test_br_tariff.py</file_path>\n<file_text>\"\"\"\nUnit tests for Brazil tariff engine (br_tariff.py).\n\nTests NCM code lookup, ICMS/IPI/PIS/COFINS calculations,\nMercosur CET rates, free zone exemptions, and edge cases.\n\"\"\"\n\nimport pytest\nfrom decimal import Decimal\nfrom datetime import date, datetime\nfrom unittest.mock import Mock, patch\n\nfrom rosetta_core.br_tariff import (\n    BrazilTariffEngine,\n    NCMCode,\n    TariffRate,\n    TaxCalculation,\n    BrazilTariffError,\n    InvalidNCMError,\n    TariffNotFoundError\n)\n\n\nclass TestNCMCode:\n    \"\"\"Test NCM (Nomenclatura Comum do Mercosul) code handling.\"\"\"\n    \n    def test_ncm_code_validation(self):\n        \"\"\"Test NCM code format validation.\"\"\"\n        # Valid 8-digit NCM codes\n        valid_codes = [\n            \"12345678\",\n            \"87032310\",  # Motor vehicles\n            \"17011100\",  # Raw sugar\n            \"02013000\"   # Fresh beef\n        ]\n        \n        for code in valid_codes:\n            ncm = NCMCode(code)\n            assert ncm.code == code\n            assert ncm.is_valid()\n    \n    def test_ncm_code_invalid_format(self):\n        \"\"\"Test invalid NCM code formats.\"\"\"\n        invalid_codes = [\n            \"1234567\",   # Too short\n            \"123456789\", # Too long\n            \"1234567a\",  # Contains letter\n            \"12.34.56.78\", # Contains dots\n            \"\",          # Empty\n            None         # None\n        ]\n        \n        for code in invalid_codes:\n            with pytest.raises(InvalidNCMError):\n                NCMCode(code)\n    \n    def test_ncm_code_normalization(self):\n        \"\"\"Test NCM code normalization.\"\"\"\n        # Test with spaces and formatting\n        ncm = NCMCode(\" 12345678 \")\n        assert ncm.code == \"12345678\"\n        \n        # Test with dots (should be removed)\n        ncm = NCMCode(\"12.34.56.78\")\n        assert ncm.code == \"12345678\"\n    \n    def test_ncm_hierarchy(self):\n        \"\"\"Test NCM code hierarchy extraction.\"\"\"\n        ncm = NCMCode(\"87032310\")\n        \n        assert ncm.chapter == \"87\"  # Chapter (first 2 digits)\n        assert ncm.heading == \"8703\"  # Heading (first 4 digits)\n        assert ncm.subheading == \"870323\"  # Subheading (first 6 digits)\n        assert ncm.item == \"87032310\"  # Full item code\n\n\nclass TestTariffRate:\n    \"\"\"Test tariff rate structure and calculations.\"\"\"\n    \n    def test_tariff_rate_creation(self):\n        \"\"\"Test basic tariff rate creation.\"\"\"\n        rate = TariffRate(\n            ncm_code=\"87032310\",\n            import_duty=Decimal(\"35.0\"),\n            ipi_rate=Decimal(\"25.0\"),\n            pis_rate=Decimal(\"1.65\"),\n            cofins_rate=Decimal(\"7.6\"),\n            effective_date=date(2024, 1, 1)\n        )\n        \n        assert rate.ncm_code == \"87032310\"\n        assert rate.import_duty == Decimal(\"35.0\")\n        assert rate.ipi_rate == Decimal(\"25.0\")\n        assert rate.pis_rate == Decimal(\"1.65\")\n        assert rate.cofins_rate == Decimal(\"7.6\")\n    \n    def test_cet_rate(self):\n        \"\"\"Test Mercosur Common External Tariff (CET) rates.\"\"\"\n        # Standard CET rate\n        rate = TariffRate(\n            ncm_code=\"17011100\",\n            import_duty=Decimal(\"16.0\"),  # Standard CET\n            is_cet=True\n        )\n        \n        assert rate.is_cet is True\n        assert rate.import_duty == Decimal(\"16.0\")\n    \n    def test_anti_dumping_duty(self):\n        \"\"\"Test anti-dumping duty calculations.\"\"\"\n        rate = TariffRate(\n            ncm_code=\"72142000\",  # Steel products\n            import_duty=Decimal(\"12.0\"),\n            anti_dumping_duty=Decimal(\"15.5\"),\n            anti_dumping_countries=[\"CN\", \"IN\"]  # China, India\n        )\n        \n        assert rate.anti_dumping_duty == Decimal(\"15.5\")\n        assert \"CN\" in rate.anti_dumping_countries\n        assert \"IN\" in rate.anti_dumping_countries\n\n\nclass TestBrazilTariffEngine:\n    \"\"\"Test main Brazil tariff engine functionality.\"\"\"\n    \n    @pytest.fixture\n    def engine(self):\n        \"\"\"Create a Brazil tariff engine instance.\"\"\"\n        return BrazilTariffEngine()\n    \n    @pytest.fixture\n    def sample_tariff_data(self):\n        \"\"\"Sample tariff data for testing.\"\"\"\n        return {\n            \"87032310\": TariffRate(  # Motor vehicles\n                ncm_code=\"87032310\",\n                import_duty=Decimal(\"35.0\"),\n                ipi_rate=Decimal(\"25.0\"),\n                pis_rate=Decimal(\"1.65\"),\n                cofins_rate=Decimal(\"7.6\"),\n                effective_date=date(2024, 1, 1)\n            ),\n            \"17011100\": TariffRate(  # Raw sugar\n                ncm_code=\"17011100\",\n                import_duty=Decimal(\"16.0\"),\n                ipi_rate=Decimal(\"0.0\"),\n                pis_rate=Decimal(\"1.65\"),\n                cofins_rate=Decimal(\"7.6\"),\n                is_cet=True\n            ),\n            \"02013000\": TariffRate(  # Fresh beef\n                ncm_code=\"02013000\",\n                import_duty=Decimal(\"0.0\"),  # Zero duty\n                ipi_rate=Decimal(\"0.0\"),\n                pis_rate=Decimal(\"1.65\"),\n                cofins_rate=Decimal(\"7.6\"),\n                quota_quantity=50000  # TRQ quota\n            )\n        }\n    \n    def test_ncm_lookup_success(self, engine, sample_tariff_data):\n        \"\"\"Test successful NCM code lookup.\"\"\"\n        engine._tariff_data = sample_tariff_data\n        \n        rate = engine.lookup_ncm(\"87032310\")\n        \n        assert rate.ncm_code == \"87032310\"\n        assert rate.import_duty == Decimal(\"35.0\")\n        assert rate.ipi_rate == Decimal(\"25.0\")\n    \n    def test_ncm_lookup_not_found(self, engine):\n        \"\"\"Test NCM code not found.\"\"\"\n        engine._tariff_data = {}\n        \n        with pytest.raises(TariffNotFoundError):\n            engine.lookup_ncm(\"99999999\")\n    \n    def test_calculate_import_duty(self, engine, sample_tariff_data):\n        \"\"\"Test import duty calculation.\"\"\"\n        engine._tariff_data = sample_tariff_data\n        \n        # Test standard duty calculation\n        duty = engine.calculate_import_duty(\n            ncm_code=\"87032310\",\n            cif_value=Decimal(\"50000.00\")\n        )\n        \n        expected = Decimal(\"50000.00\") * Decimal(\"35.0\") / 100\n        assert duty == expected  # 17,500.00\n    \n    def test_calculate_ipi(self, engine, sample_tariff_data):\n        \"\"\"Test IPI (Imposto sobre Produtos Industrializados) calculation.\"\"\"\n        engine._tariff_data = sample_tariff_data\n        \n        cif_value = Decimal(\"50000.00\")\n        import_duty = Decimal(\"17500.00\")\n        \n        ipi = engine.calculate_ipi(\n            ncm_code=\"87032310\",\n            cif_value=cif_value,\n            import_duty=import_duty\n        )\n        \n        # IPI base = CIF + Import Duty\n        ipi_base = cif_value + import_duty\n        expected = ipi_base * Decimal(\"25.0\") / 100\n        assert ipi == expected  # 16,875.00\n    \n    def test_calculate_pis_cofins(self, engine, sample_tariff_data):\n        \"\"\"Test PIS/COFINS calculation.\"\"\"\n        engine._tariff_data = sample_tariff_data\n        \n        cif_value = Decimal(\"50000.00\")\n        import_duty = Decimal(\"17500.00\")\n        \n        pis = engine.calculate_pis(\n            ncm_code=\"87032310\",\n            cif_value=cif_value,\n            import_duty=import_duty\n        )\n        \n        cofins = engine.calculate_cofins(\n            ncm_code=\"87032310\",\n            cif_value=cif_value,\n            import_duty=import_duty\n        )\n        \n        # PIS/COFINS base = CIF + Import Duty\n        base = cif_value + import_duty\n        expected_pis = base * Decimal(\"1.65\") / 100\n        expected_cofins = base * Decimal(\"7.6\") / 100\n        \n        assert pis == expected_pis\n        assert cofins == expected_cofins\n    \n    def test_calculate_icms(self, engine):\n        \"\"\"Test ICMS (state tax) calculation.\"\"\"\n        # ICMS rates vary by state\n        icms_rate = Decimal(\"18.0\")  # São Paulo standard rate\n        \n        # ICMS calculation includes all other taxes in base\n        cif_value = Decimal(\"50000.00\")\n        import_duty = Decimal(\"17500.00\")\n        ipi = Decimal(\"16875.00\")\n        pis = Decimal(\"1113.75\")\n        cofins = Decimal(\"5130.00\")\n        \n        # ICMS base = (CIF + II + IPI + PIS + COFINS) / (1 - ICMS_rate)\n        total_without_icms = cif_value + import_duty + ipi + pis + cofins\n        icms_base = total_without_icms / (1 - icms_rate / 100)\n        icms = icms_base - total_without_icms\n        \n        calculated_icms = engine.calculate_icms(\n            cif_value=cif_value,\n            import_duty=import_duty,\n            ipi=ipi,\n            pis=pis,\n            cofins=cofins,\n            icms_rate=icms_rate\n        )\n        \n        assert abs(calculated_icms - icms) < Decimal(\"0.01\")\n    \n    def test_manaus_free_zone_exemption(self, engine, sample_tariff_data):\n        \"\"\"Test Manaus Free Zone exemptions.\"\"\"\n        engine._tariff_data = sample_tariff_data\n        \n        # Eligible product for Manaus exemption\n        calculation = engine.calculate_total_taxes(\n            ncm_code=\"87032310\",\n            cif_value=Decimal(\"50000.00\"),\n            destination_state=\"AM\",  # Amazonas (Manaus)\n            free_zone_eligible=True\n        )\n        \n        # Import duty should be exempt for eligible products\n        assert calculation.import_duty == Decimal(\"0.00\")\n        # IPI should also be exempt\n        assert calculation.ipi == Decimal(\"0.00\")\n        # PIS/COFINS may still apply\n        assert calculation.pis > Decimal(\"0.00\")\n        assert calculation.cofins > Decimal(\"0.00\")\n    \n    def test_mercosur_origin_preference(self, engine, sample_tariff_data):\n        \"\"\"Test Mercosur origin preferences.\"\"\"\n        engine._tariff_data = sample_tariff_data\n        \n        # Product from Argentina (Mercosur member)\n        calculation = engine.calculate_total_taxes(\n            ncm_code=\"02013000\",  # Beef\n            cif_value=Decimal(\"10000.00\"),\n            origin_country=\"AR\",  # Argentina\n            has_origin_certificate=True\n        )\n        \n        # Should have preferential (zero) duty rate\n        assert calculation.import_duty == Decimal(\"0.00\")\n    \n    def test_temporary_tariff_reduction(self, engine, sample_tariff_data):\n        \"\"\"Test temporary tariff reductions.\"\"\"\n        # Modify tariff data to include temporary reduction\n        temp_rate = sample_tariff_data[\"87032310\"]\n        temp_rate.temporary_duty = Decimal(\"20.0\")  # Reduced from 35%\n        temp_rate.temp_duty_start = date(2024, 1, 1)\n        temp_rate.temp_duty_end = date(2024, 12, 31)\n        \n        engine._tariff_data = sample_tariff_data\n        \n        # Calculate with temporary rate\n        with patch('rosetta_core.br_tariff.date') as mock_date:\n            mock_date.today.return_value = date(2024, 6, 1)  # Within temp period\n            \n            duty = engine.calculate_import_duty(\n                ncm_code=\"87032310\",\n                cif_value=Decimal(\"50000.00\"),\n                apply_temporary_rates=True\n            )\n            \n            expected = Decimal(\"50000.00\") * Decimal(\"20.0\") / 100\n            assert duty == expected  # 10,000.00 (reduced rate)\n    \n    def test_anti_dumping_duty_application(self, engine):\n        \"\"\"Test anti-dumping duty application.\"\"\"\n        # Add anti-dumping rate\n        rate = TariffRate(\n            ncm_code=\"72142000\",\n            import_duty=Decimal(\"12.0\"),\n            anti_dumping_duty=Decimal(\"15.5\"),\n            anti_dumping_countries=[\"CN\"]\n        )\n        \n        engine._tariff_data = {\"72142000\": rate}\n        \n        # Calculate for Chinese origin (subject to anti-dumping)\n        calculation = engine.calculate_total_taxes(\n            ncm_code=\"72142000\",\n            cif_value=Decimal(\"30000.00\"),\n            origin_country=\"CN\"\n        )\n        \n        # Should include both regular duty and anti-dumping duty\n        expected_regular = Decimal(\"30000.00\") * Decimal(\"12.0\") / 100\n        expected_ad = Decimal(\"30000.00\") * Decimal(\"15.5\") / 100\n        \n        assert calculation.import_duty == expected_regular + expected_ad\n    \n    def test_quota_system(self, engine, sample_tariff_data):\n        \"\"\"Test tariff rate quota (TRQ) system.\"\"\"\n        engine._tariff_data = sample_tariff_data\n        \n        # Simulate quota usage\n        engine._quota_usage = {\"02013000\": 45000}  # 45,000 tons used\n        \n        # Within quota - should get preferential rate\n        calc_within = engine.calculate_total_taxes(\n            ncm_code=\"02013000\",\n            cif_value=Decimal(\"10000.00\"),\n            quantity_kg=3000000  # 3,000 tons (still within quota)\n        )\n        \n        assert calc_within.import_duty == Decimal(\"0.00\")  # In-quota rate\n        \n        # Exceed quota - should get out-of-quota rate\n        calc_exceed = engine.calculate_total_taxes(\n            ncm_code=\"02013000\",\n            cif_value=Decimal(\"10000.00\"),\n            quantity_kg=8000000,  # 8,000 tons (exceeds remaining quota)\n            out_quota_rate=Decimal(\"55.0\")\n        )\n        \n        # Should apply out-of-quota rate to excess quantity\n        assert calc_exceed.import_duty > Decimal(\"0.00\")\n    \n    def test_error_handling(self, engine):\n        \"\"\"Test error handling for various scenarios.\"\"\"\n        # Invalid NCM code\n        with pytest.raises(InvalidNCMError):\n            engine.lookup_ncm(\"invalid\")\n        \n        # NCM not found\n        with pytest.raises(TariffNotFoundError):\n            engine.lookup_ncm(\"99999999\")\n        \n        # Invalid calculation parameters\n        with pytest.raises(BrazilTariffError):\n            engine.calculate_import_duty(\n                ncm_code=\"87032310\",\n                cif_value=Decimal(\"-1000.00\")  # Negative value\n            )\n\n\nclass TestTaxCalculation:\n    \"\"\"Test tax calculation result structure.\"\"\"\n    \n    def test_tax_calculation_creation(self):\n        \"\"\"Test tax calculation result creation.\"\"\"\n        calc = TaxCalculation(\n            ncm_code=\"87032310\",\n            cif_value=Decimal(\"50000.00\"),\n            import_duty=Decimal(\"17500.00\"),\n            ipi=Decimal(\"16875.00\"),\n            pis=Decimal(\"1113.75\"),\n            cofins=Decimal(\"5130.00\"),\n            icms=Decimal(\"16097.56\"),\n            total_taxes=Decimal(\"56716.31\")\n        )\n        \n        assert calc.ncm_code == \"87032310\"\n        assert calc.total_taxes == Decimal(\"56716.31\")\n    \n    def test_effective_tax_rate(self):\n        \"\"\"Test effective tax rate calculation.\"\"\"\n        calc = TaxCalculation(\n            ncm_code=\"87032310\",\n            cif_value=Decimal(\"50000.00\"),\n            import_duty=Decimal(\"17500.00\"),\n            ipi=Decimal(\"16875.00\"),\n            pis=Decimal(\"1113.75\"),\n            cofins=Decimal(\"5130.00\"),\n            icms=Decimal(\"16097.56\"),\n            total_taxes=Decimal(\"56716.31\")\n        )\n        \n        effective_rate = calc.effective_tax_rate()\n        expected_rate = (Decimal(\"56716.31\") / Decimal(\"50000.00\")) * 100\n        \n        assert abs(effective_rate - expected_rate) < Decimal(\"0.01\")\n    \n    def test_tax_breakdown(self):\n        \"\"\"Test tax breakdown by type.\"\"\"\n        calc = TaxCalculation(\n            ncm_code=\"87032310\",\n            cif_value=Decimal(\"50000.00\"),\n            import_duty=Decimal(\"17500.00\"),\n            ipi=Decimal(\"16875.00\"),\n            pis=Decimal(\"1113.75\"),\n            cofins=Decimal(\"5130.00\"),\n            icms=Decimal(\"16097.56\"),\n            total_taxes=Decimal(\"56716.31\")\n        )\n        \n        breakdown = calc.tax_breakdown()\n        \n        assert \"import_duty\" in breakdown\n        assert \"ipi\" in breakdown\n        assert \"pis\" in breakdown\n        assert \"cofins\" in breakdown\n        assert \"icms\" in breakdown\n        \n        # Verify percentages sum to 100%\n        total_percentage = sum(breakdown.values())\n        assert abs(total_percentage - 100) < 0.01\n\n\nclass TestEdgeCases:\n    \"\"\"Test edge cases and special scenarios.\"\"\"\n    \n    def test_zero_value_import(self):\n        \"\"\"Test handling of zero-value imports.\"\"\"\n        engine = BrazilTariffEngine()\n        \n        with pytest.raises(BrazilTariffError, match=\"CIF value must be positive\"):\n            engine.calculate_total_taxes(\n                ncm_code=\"87032310\",\n                cif_value=Decimal(\"0.00\")\n            )\n    \n    def test_maximum_duty_cap(self):\n        \"\"\"Test maximum duty caps for certain products.\"\"\"\n        engine = BrazilTariffEngine()\n        \n        # Some products have maximum duty caps\n        rate = TariffRate(\n            ncm_code=\"30049099\",  # Medicines\n            import_duty=Decimal(\"14.0\"),\n            max_duty_cap=Decimal(\"1.00\")  # R$ 1.00 per kg max\n        )\n        \n        engine._tariff_data = {\"30049099\": rate}\n        \n        # High-value, low-weight scenario where cap applies\n        calculation = engine.calculate_total_taxes(\n            ncm_code=\"30049099\",\n            cif_value=Decimal(\"100000.00\"),\n            quantity_kg=50.0  # 50 kg\n        )\n        \n        # Duty should be capped at R$ 50.00 (50 kg × R$ 1.00/kg)\n        assert calculation.import_duty == Decimal(\"50.00\")\n    \n    def test_date_sensitive_rates(self):\n        \"\"\"Test date-sensitive tariff rates.\"\"\"\n        engine = BrazilTariffEngine()\n        \n        # Rate that changed over time\n        old_rate = TariffRate(\n            ncm_code=\"87032310\",\n            import_duty=Decimal(\"35.0\"),\n            effective_date=date(2023, 1, 1),\n            expiry_date=date(2023, 12, 31)\n        )\n        \n        new_rate = TariffRate(\n            ncm_code=\"87032310\",\n            import_duty=Decimal(\"20.0\"),\n            effective_date=date(2024, 1, 1)\n        )\n        \n        engine._tariff_data = {\"87032310\": [old_rate, new_rate]}\n        \n        # Should use rate effective for the calculation date\n        with patch('rosetta_core.br_tariff.date') as mock_date:\n            mock_date.today.return_value = date(2024, 6, 1)\n            \n            rate = engine.lookup_ncm(\"87032310\")\n            assert rate.import_duty == Decimal(\"20.0\")\n    \n    def test_complex_calculation_chain(self):\n        \"\"\"Test complex calculation with all taxes and exemptions.\"\"\"\n        engine = BrazilTariffEngine()\n        \n        rate = TariffRate(\n            ncm_code=\"84713012\",  # Computers\n            import_duty=Decimal(\"16.0\"),\n            ipi_rate=Decimal(\"0.0\"),\n            pis_rate=Decimal(\"1.65\"),\n            cofins_rate=Decimal(\"7.6\"),\n            is_it_goods=True  # IT goods with special treatment\n        )\n        \n        engine._tariff_data = {\"84713012\": rate}\n        \n        calculation = engine.calculate_total_taxes(\n            ncm_code=\"84713012\",\n            cif_value=Decimal(\"75000.00\"),\n            origin_country=\"KR\",  # South Korea\n            destination_state=\"SP\",  # São Paulo\n            icms_rate=Decimal(\"18.0\"),\n            has_ex_tariff=True  # Ex-tariff benefit\n        )\n        \n        # With Ex-tariff, import duty should be reduced/eliminated\n        assert calculation.import_duty < rate.import_duty * Decimal(\"75000.00\") / 100\n        assert calculation.total_taxes > Decimal(\"0.00\")\n        assert isinstance(calculation.breakdown_by_tax_type(), dict)\n\n\nif __name__ == \"__main__\":\n    pytest.main([__file__])\n</file_text>\n\nI've created comprehensive unit tests for the Brazil tariff engine covering all the requirements:\n\n## Key Test Areas Covered:\n\n### 1. **NCM Code Handling**\n- Validation of 8-digit NCM codes\n- Code normalization and formatting\n- Hierarchy extraction (chapter, heading, subheading, item)\n- Invalid format handling\n\n### 2. **Tax Calculations**\n- **Import Duty**: Standard rates, temporary reductions\n- **IPI** (Industrialized Products Tax): Based on CIF + duty\n- **PIS/COFINS**: Federal social contributions \n- **ICMS**: State tax with complex base calculation\n\n### 3. **Mercosur CET (Common External Tariff)**\n- Standard CET rates\n- Mercosur origin preferences\n- Certificate of origin validation\n\n### 4. **Free Zone Exemptions**\n- Manaus Free Zone (Amazonas state)\n- Eligible product exemptions\n- Duty and IPI exemptions\n\n### 5. **Edge Cases & Special Scenarios**\n- **Anti-dumping duties**: Country-specific additional duties\n- **Temporary tariff reductions**: Time-limited rate changes\n- **Tariff Rate Quotas (TRQ)**: In-quota vs out-of-quota rates\n- **Maximum duty caps**: For medicines and special products\n- **Ex-tariff benefits**: Duty reductions for certain imports\n- **Date-sensitive rates**: Historical rate changes\n\n### 6. **Error Handling**\n- Invalid NCM codes\n- Tariff not found scenarios\n- Negative values and validation\n- Complex calculation chains\n\n### 7. **Calculation Results**\n- Tax breakdown by type\n- Effective tax rate calculation\n- Total tax computation\n- Percentage analysis\n\nThe tests use proper fixtures, mocking for date-sensitive operations, and comprehensive assertions to ensure the tariff engine works correctly across all scenarios commonly encountered in Brazilian import operations."
}