def test_to_praat_starts_before(self):
mock_tier = mock.Mock()
mock_tier.xmin = 0
mock_tier.xmax = 1
tg = TextGrid(0.5, 1, [mock_tier])
with assert_raises(ValueError):
tg.to_praat()
def test_to_praat_ends_after(self):
mock_tier = mock.Mock()
mock_tier.xmin = 0
mock_tier.xmax = 1
tg = TextGrid(0, 0.5, [mock_tier])
with assert_raises(ValueError):
tg.to_praat()
def test_to_praat(self):
mock_tier = mock.Mock()
mock_tier.xmin = 0
mock_tier.xmax = 1
mock_tier.to_praat.return_value = 'a tier'
tg = TextGrid(0, 1, [mock_tier, mock_tier])
res = tg.to_praat()
assert_equal(res, ('"ooTextFile"\n"TextGrid"\n'
'0 to 1 seconds <exists>\n'
'2 tiers\n\na tier\n\na tier'))
def test_to_dict(self):
mock_tier = mock.Mock()
mock_tier.to_dict.return_value = {'a tier': 'data'}
tg = TextGrid(0, 1, [mock_tier])
res = tg.to_dict()
assert_equal(res, {
'xmin': 0,
'xmax': 1,
'tiers': [{
'a tier': 'data'
}]
})
def test_to_praat_with_path(self):
mock_tier = mock.Mock()
mock_tier.xmin = 0
mock_tier.xmax = 1
mock_tier.to_praat.return_value = 'a tier'
tg = TextGrid(0, 1, [mock_tier, mock_tier])
tg.to_praat('test/files/output.TextGrid')
with open('test/files/output.TextGrid') as output_file:
res = output_file.read()
assert_equal(res, ('"ooTextFile"\n"TextGrid"\n'
'0 to 1 seconds <exists>\n'
'2 tiers\n\na tier\n\na tier'))
def test_iter(self):
tg = TextGrid(0, 1, ['tier 0', 'tier 1'])
assert_equal(list(iter(tg)), ['tier 0', 'tier 1'])
def test_reversed(self):
tg = TextGrid(0, 1, ['tier 0', 'tier 1'])
assert_equal(list(reversed(tg)), ['tier 1', 'tier 0'])
def test_len(self):
tg = TextGrid(0, 1, ['tier 0', 'tier 1'])
assert_equal(len(tg), 2)
def test_del(self):
tg = TextGrid(0, 1, ['tier 0', 'tier 1'])
del tg[0]
assert_equal(tg[0], 'tier 1')
def test_set(self):
tg = TextGrid(0, 1, ['tier 0', 'tier 1'])
tg[1] = 'new tier'
assert_equal(tg[1], 'new tier')
def test_get(self):
tg = TextGrid(0, 1, ['tier 0', 'tier 1'])
assert_equal(tg[1], 'tier 1')
def test_str(self):
tg = TextGrid(0, 1, [1, 2, 3])
assert_equal(str(tg), '<TextGrid from 0 to 1 seconds with 3 tiers>')
def test_repr(self):
tg = TextGrid(0, 1, [1, 2, 3])
assert_equal(repr(tg), "TextGrid(0, 1, [1, 2, 3])")
def test_init(self):
tg = TextGrid(0, 1, [])
assert_equal(tg.xmin, 0)
assert_equal(tg.xmax, 1)
assert_equal(tg.tiers, [])
