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