Python Testing dengan pytest

# test_keranjang.py — Tulis test DULU (akan gagal karena class belum ada)
import pytest
from keranjang import Keranjang, Produk

class TestKeranjang:
    @pytest.fixture
    def keranjang(self):
        return Keranjang()

    @pytest.fixture
    def produk_laptop(self):
        return Produk(id=1, nama="Laptop", harga=10000000)

    @pytest.fixture
    def produk_mouse(self):
        return Produk(id=2, nama="Mouse", harga=150000)

    def test_keranjang_awalnya_kosong(self, keranjang):
        assert keranjang.jumlah_item() == 0
        assert keranjang.total() == 0

    def test_tambah_produk(self, keranjang, produk_laptop):
        keranjang.tambah(produk_laptop)
        assert keranjang.jumlah_item() == 1

    def test_tambah_produk_dengan_jumlah(self, keranjang, produk_mouse):
        keranjang.tambah(produk_mouse, jumlah=3)
        assert keranjang.jumlah_item() == 3

    def test_total_harga(self, keranjang, produk_laptop, produk_mouse):
        keranjang.tambah(produk_laptop)
        keranjang.tambah(produk_mouse, jumlah=2)
        assert keranjang.total() == 10000000 + (150000 * 2)

    def test_hapus_produk(self, keranjang, produk_laptop):
        keranjang.tambah(produk_laptop)
        keranjang.hapus(produk_laptop.id)
        assert keranjang.jumlah_item() == 0

    def test_hapus_produk_tidak_ada(self, keranjang):
        with pytest.raises(ValueError, match="Produk tidak ditemukan"):
            keranjang.hapus(999)

    def test_produk_string(self, produk_laptop):
        assert str(produk_laptop) == "Laptop (Rp 10,000,000)"

# keranjang.py — Implementasi agar test lulus
class Produk:
    def __init__(self, id, nama, harga):
        self.id = id
        self.nama = nama
        self.harga = harga

    def __str__(self):
        return f"{self.nama} (Rp {self.harga:,.0f})"

class Keranjang:
    def __init__(self):
        self._items = {}  # {produk_id: {'produk': ..., 'jumlah': ...}}

    def tambah(self, produk, jumlah=1):
        if produk.id in self._items:
            self._items[produk.id]['jumlah'] += jumlah
        else:
            self._items[produk.id] = {'produk': produk, 'jumlah': jumlah}

    def hapus(self, produk_id):
        if produk_id not in self._items:
            raise ValueError("Produk tidak ditemukan")
        del self._items[produk_id]

    def jumlah_item(self):
        return sum(item['jumlah'] for item in self._items.values())

    def total(self):
        return sum(
            item['produk'].harga * item['jumlah']
            for item in self._items.values()
        )

# Menjalankan:
# pytest test_keranjang.py -v
# Semua 7 test seharusnya PASSED ✅