python-escpos/test/test_profile.py

import pytest
from escpos.capabilities import get_profile, NotSupported, BARCODE_B, Profile


@pytest.fixture
def profile():
    return get_profile('default')


class TestBaseProfile:
    """Test the `BaseProfile` class.
    """

    def test_get_font(self, profile):
        with pytest.raises(NotSupported):
            assert profile.get_font('3')
        assert profile.get_font(1) == 1
        assert profile.get_font('a') == 0

    def test_supports(self, profile):
        assert not profile.supports('asdf asdf')
        assert profile.supports(BARCODE_B)

    def test_get_columns(self, profile):
        assert profile.get_columns('a') > 5
        with pytest.raises(NotSupported):
            assert profile.get_columns('asdfasdf')


class TestCustomProfile:
    """Test custom profile options with the `Profile` class.
    """

    def test_columns(self):
        assert Profile(columns=10).get_columns('sdfasdf') == 10

    def test_features(self):
        assert Profile(features={'foo': True}).supports('foo')