File size: 532 Bytes
d942a9b
 
2d76b56
d942a9b
 
 
 
 
 
 
 
 
 
 
 
 
2d76b56
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
"""Test radio_embed."""
# pylint: disable=broad-except
from radio_embed import __version__, radio_embed


def test_version():
    """Test version."""
    assert __version__[:3] == "0.1"


def test_sanity():
    """Check sanity."""
    try:
        assert not radio_embed()
    except Exception:
        assert True


def test_radio_embed_two_three_lines():
    """Test radio_embed two three lines."""
    res = radio_embed("a\n b")
    assert res.shape == (2, 512)

    res = radio_embed("a\n \nb").shape
    assert res == (3, 512)