1from translate.lang import factory
2
3
4def test_punctranslate():
5    """Tests that we can translate punctuation."""
6    language = factory.getlanguage("hy")
7    assert language.punctranslate("") == ""
8    assert language.punctranslate("abc efg") == "abc efg"
9    assert language.punctranslate("abc efg.") == "abc efg։"
10    assert language.punctranslate("abc efg. hij.") == "abc efg։ hij։"
11    assert language.punctranslate("abc efg!") == "abc efg՜"
12    assert language.punctranslate("Delete file: %s") == "Delete file՝ %s"
13    # TODO: Find out exactly how questions work
14
15
16def test_sentences():
17    """Tests basic functionality of sentence segmentation."""
18    language = factory.getlanguage("hy")
19    sentences = language.sentences("")
20    assert sentences == []
21
22    sentences = language.sentences(
23        "Արխիվն արդեն գոյություն ունի։ Դուք ցանկանու՞մ եք կրկին գրել այն։"
24    )
25    assert sentences == [
26        "Արխիվն արդեն գոյություն ունի։",
27        "Դուք ցանկանու՞մ եք կրկին գրել այն։",
28    ]
29    sentences = language.sentences(
30        "Արխիվն արդեն գոյություն ունի։ դուք ցանկանու՞մ եք կրկին գրել այն։"
31    )
32    assert sentences == [
33        "Արխիվն արդեն գոյություն ունի։ դուք ցանկանու՞մ եք կրկին գրել այն։"
34    ]
35