Commit f8ef1c8

committed

updated a function in Point and added some tests

1 parent 6836ee9 commit f8ef1c8Copy full SHA for f8ef1c8

File tree

+33

-4

lines changed

+33

-4

lines changed

Lines changed: 4 additions & 4 deletions

Original file line number	Diff line number	Diff line change
`@@ -50,10 +50,6 @@ def get_state(self):`
`50`	`50`	`def print_state(self):`
`51`	`51`	`return " " if self.state.name == "E" else self.state.name`
`52`	`52`
`53`		`- @property`
`54`		`- def coord(self):`
`55`		`- return self._index_to_coord[self.index]`
`56`		`-`
`57`	`53`	`@classmethod`
`58`	`54`	`def get_index(cls, coord):`
`59`	`55`	`assert (`
`@@ -66,6 +62,10 @@ def get_coord(cls, index):`
`66`	`62`	`assert 0 <= index < 25, "Invalid index: %d" % index`
`67`	`63`	`return cls._index_to_coord[index]`
`68`	`64`
	`65`	`+ @property`
	`66`	`+ def coord(self):`
	`67`	`+ return Point.get_coord(self.index)`
	`68`	`+`
`69`	`69`	`def __str__(self):`
`70`	`70`	`return self.coord`
`71`	`71`

Lines changed: 29 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,29 @@`
	`1`	`+import unittest`
	`2`	`+`
	`3`	`+from Point import Point`
	`4`	`+`
	`5`	`+`
	`6`	`+class TestPoint(unittest.TestCase):`
	`7`	`+ def setUp(self):`
	`8`	`+ self.points = [Point(i) for i in range(25)]`
	`9`	`+`
	`10`	`+ def test_coord(self):`
	`11`	`+ self.assertEqual(self.points[0].coord, 'A1')`
	`12`	`+ self.assertEqual(self.points[12].coord, 'C3')`
	`13`	`+ self.assertEqual(self.points[24].coord, 'E5')`
	`14`	`+`
	`15`	`+ def test_index(self):`
	`16`	`+ self.assertEqual(self.points[0].index, 0)`
	`17`	`+ self.assertEqual(self.points[12].index, 12)`
	`18`	`+`
	`19`	`+ def test_get_index(self):`
	`20`	`+ self.assertEqual(Point.get_index('C3'), 12)`
	`21`	`+ self.assertEqual(Point.get_index('E4'), 19)`
	`22`	`+`
	`23`	`+ def test_set_state(self):`
	`24`	`+ self.points[0].set_state('T')`
	`25`	`+ self.assertEqual(self.points[0].state, Point.State.T)`
	`26`	`+ self.assertEqual(self.points[0].get_state(), Point.State.T)`
	`27`	`+`
	`28`	`+ def test_str(self):`
	`29`	`+ self.assertEqual(str(self.points[0]), 'A1')`

Comments

(0)