cell_loader_ctrl/robot-control/tests/test_vision.py

import pytest
from vision.datamatrix import DataMatrixReader
import numpy as np

@pytest.fixture
def datamatrix_reader():
    return DataMatrixReader(-1)
@pytest.fixture
def datamatrix_reader_realcam():
    return DataMatrixReader()

class TestDataMatrixReader:
    def test_init_reader(self, datamatrix_reader_realcam: DataMatrixReader):
        assert datamatrix_reader_realcam is not None

    def test_read_success(self, datamatrix_reader: DataMatrixReader):
        datamatrix_reader.initialize()
        result = datamatrix_reader.read_datamatrix()
        assert result == "https://batteries.up-cell.de/cells/2224"

    def test_read_failure(self, datamatrix_reader: DataMatrixReader):
        datamatrix_reader.initialize()
        result = datamatrix_reader.read_datamatrix() # first is success
        result = datamatrix_reader.read_datamatrix() # second is failure
        assert result is None

    def test_cleanup(self, datamatrix_reader: DataMatrixReader):
        datamatrix_reader.initialize()
        datamatrix_reader.cleanup()
        assert datamatrix_reader.camera is not None
        
    def test_cleanup_real_camera(self, datamatrix_reader_realcam: DataMatrixReader):
        try:
            datamatrix_reader_realcam.initialize()
            datamatrix_reader_realcam.cleanup()
            assert datamatrix_reader_realcam.camera is not None
        except Exception:
            pytest.skip("No real camera available for testing")