def test_track_str(self):
self.assertEqual('CHANX 0,0->10,0', str(Track((0, 0), (10, 0))))
self.assertEqual('CHANY 0,0->0,10', str(Track((0, 0), (0, 10))))
self.assertEqual('CHANX 1,2->3,2 @5', str(Track((1, 2), (3, 2),
idx=5)))
self.assertEqual('ABC', str(Track((1, 2), (3, 2), name="ABC")))
self.assertEqual('ABC@5', str(Track((1, 2), (3, 2), idx=5,
name="ABC")))
def test_track_repr(self):
self.assertEqual('T((0,0), (10,0))', repr(Track((0, 0), (10, 0))))
self.assertEqual('T((0,0), (0,10))', repr(Track((0, 0), (0, 10))))
self.assertEqual('T((1,2), (3,2), 5)',
repr(Track((1, 2), (3, 2), idx=5)))
self.assertEqual('T(ABC)', repr(Track((1, 2), (3, 2), name="ABC")))
self.assertEqual('T(ABC,5)',
repr(Track((1, 2), (3, 2), idx=5, name="ABC")))
def test_track_new_idx(self):
s = (1, 4)
e = (1, 8)
c1 = Track(s, e, idx=0)
c2 = c1.new_idx(2)
self.assertEqual(c1.start, c2.start)
self.assertEqual(c1.end, c2.end)
self.assertEqual(0, c1.idx)
self.assertEqual(2, c2.idx)
def test_track_type(self):
self.assertEqual(Track.Type.X, Track((1, 0), (10, 0)).type)
self.assertEqual(Track.Type.Y, Track((0, 1), (0, 10)).type)
with self.assertRaises(ValueError):
Track((1, 1), (1, 1)).type
self.assertEqual(Track.Type.X,
Track((1, 1), (1, 1), type_hint=Track.Type.X).type)
self.assertEqual(Track.Type.Y,
Track((1, 1), (1, 1), type_hint=Track.Type.Y).type)
def test_track_common(self):
self.assertEqual(0, Track((0, 0), (10, 0)).common)
self.assertEqual(0, Track((0, 0), (0, 10)).common)
with self.assertRaises(ValueError):
Track((1, 1), (1, 1)).common
self.assertEqual(1,
Track((1, 1), (1, 1), type_hint=Track.Type.X).common)
self.assertEqual(0, Track((10, 0), (0, 0)).common)
self.assertEqual(0, Track((0, 10), (0, 0)).common)
self.assertEqual(4, Track((4, 10), (4, 0)).common)
def test_channelgrid_create_track(self):
g = ChannelGrid((11, 11), Track.Type.X)
# Adding the first channel
g.create_track(Track((1, 6), (4, 6), name="A"))
self.assertEqual('[T(A,0)]', str(g[(1, 6)]))
self.assertEqual('[T(A,0)]', str(g[(2, 6)]))
self.assertEqual('[T(A,0)]', str(g[(4, 6)]))
self.assertEqual('[None]', str(g[(5, 6)]))
# Adding second non-overlapping second channel
g.create_track(Track((5, 6), (7, 6), name="B"))
self.assertEqual('[T(A,0)]', str(g[(4, 6)]))
self.assertEqual('[T(B,0)]', str(g[(5, 6)]))
self.assertEqual('[T(B,0)]', str(g[(7, 6)]))
self.assertEqual('[None]', str(g[(8, 6)]))
# Adding third channel which overlaps with second channel
g.create_track(Track((5, 6), (7, 6), name="T"))
self.assertEqual('[T(A,0), None]', str(g[(4, 6)]))
self.assertEqual('[T(B,0), T(T,1)]', str(g[(5, 6)]))
self.assertEqual('[T(B,0), T(T,1)]', str(g[(7, 6)]))
# Adding a channel which overlaps, but is a row over
g.create_track(Track((5, 7), (7, 7), name="D"))
self.assertEqual('[T(B,0), T(T,1)]', str(g[(5, 6)]))
self.assertEqual('[T(D,0)]', str(g[(5, 7)]))
# Adding fourth channel which overlaps both the first
# and second+third channel
g.create_track(Track((3, 6), (6, 6), name="E"))
self.assertEqual('[T(A,0), None, None]', str(g[(2, 6)]))
self.assertEqual('[T(A,0), None, T(E,2)]', str(g[(3, 6)]))
self.assertEqual('[T(B,0), T(T,1), T(E,2)]', str(g[(6, 6)]))
self.assertEqual('[T(B,0), T(T,1), None]', str(g[(7, 6)]))
# This channel fits in the hole left by the last one.
g.create_track(Track((1, 6), (3, 6), name="F"))
self.assertEqual('[T(A,0), T(F,1), None]', str(g[(1, 6)]))
self.assertEqual('[T(A,0), T(F,1), None]', str(g[(2, 6)]))
self.assertEqual('[T(A,0), T(F,1), T(E,2)]', str(g[(3, 6)]))
self.assertEqual('[T(A,0), None, T(E,2)]', str(g[(4, 6)]))
# Add another channel which causes a hole
g.create_track(Track((1, 6), (7, 6), name="G"))
self.assertEqual('[T(A,0), T(F,1), None, T(G,3)]', str(g[(1, 6)]))
self.assertEqual('[T(A,0), T(F,1), None, T(G,3)]', str(g[(2, 6)]))
self.assertEqual('[T(A,0), T(F,1), T(E,2), T(G,3)]', str(g[(3, 6)]))
self.assertEqual('[T(A,0), None, T(E,2), T(G,3)]', str(g[(4, 6)]))
self.assertEqual('[T(B,0), T(T,1), T(E,2), T(G,3)]', str(g[(5, 6)]))
self.assertEqual('[T(B,0), T(T,1), T(E,2), T(G,3)]', str(g[(6, 6)]))
self.assertEqual('[T(B,0), T(T,1), None, T(G,3)]', str(g[(7, 6)]))
self.assertEqual('[None, None, None, None]', str(g[(8, 6)]))
def test_track_end0(self):
self.assertEqual(10, Track((1, 0), (10, 0)).end0)
self.assertEqual(10, Track((0, 1), (0, 10)).end0)
with self.assertRaises(ValueError):
Track((1, 1), (1, 1)).end0
self.assertEqual(1, Track((1, 1), (1, 1), type_hint=Track.Type.X).end0)
self.assertEqual(1, Track((10, 0), (1, 0)).end0)
self.assertEqual(1, Track((0, 10), (0, 1)).end0)
def test_track_length(self):
self.assertEqual(10, Track((0, 0), (10, 0)).length)
self.assertEqual(10, Track((0, 0), (0, 10)).length)
self.assertEqual(0, Track((1, 1), (1, 1)).length)
self.assertEqual(10, Track((10, 0), (0, 0)).length)
self.assertEqual(10, Track((0, 10), (0, 0)).length)
def test_track_type_guess(self):
self.assertEqual(Track.Type.X, Track((1, 0), (10, 0)).type_guess)
self.assertEqual(Track.Type.Y, Track((0, 1), (0, 10)).type_guess)
self.assertEqual('None', str(Track((1, 1), (1, 1)).type_guess))