175 lines
5.0 KiB
Python
175 lines
5.0 KiB
Python
from unittest import TestCase
|
|
|
|
from gcodeextractor.gcode.path import scanHorizontal, scanVertical, twoRemoving
|
|
|
|
|
|
class TestPath(TestCase):
|
|
|
|
#inputs
|
|
def imageTest(self):
|
|
return [[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 0, 0, 1, 0, 0, 0],
|
|
[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 0, 0, 0, 0, 0, 0]]
|
|
|
|
def imageTestMulti(self):
|
|
return [[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 1, 0, 0, 1, 0, 0],
|
|
[0, 0, 0, 1, 0, 0, 0],
|
|
[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 0, 0, 0, 1, 0, 0]]
|
|
|
|
def imageTestHOut(self):
|
|
return [[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 0, 0, 1, 0, 0, 0],
|
|
[0, 0, 0, 0, 0, 0, 0]]
|
|
|
|
def imageTestVOut(self):
|
|
return [[0, 0, 0],
|
|
[0, 0, 0],
|
|
[0, 0, 0],
|
|
[0, 1, 0],
|
|
[0, 0, 0],
|
|
[0, 0, 0],
|
|
[0, 0, 0]]
|
|
|
|
def imageTest2(self):
|
|
return [[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 2, 2, 2, 2, 2, 0],
|
|
[0, 2, 2, 2, 2, 2, 0],
|
|
[0, 2, 2, 1, 2, 2, 0],
|
|
[0, 2, 2, 2, 2, 2, 0],
|
|
[0, 2, 2, 2, 2, 2, 0],
|
|
[0, 0, 0, 0, 0, 0, 0]]
|
|
|
|
def imageTestMulti2(self):
|
|
return [[2, 2, 2, 2, 2, 2, 2],
|
|
[2, 2, 2, 2, 2, 2, 2],
|
|
[2, 1, 2, 2, 1, 2, 2],
|
|
[2, 2, 2, 1, 2, 2, 2],
|
|
[2, 2, 2, 2, 2, 2, 2],
|
|
[2, 2, 2, 2, 2, 2, 2],
|
|
[2, 2, 2, 2, 1, 2, 2]]
|
|
|
|
|
|
#horizontal results
|
|
def imageResultHr2(self):
|
|
return [[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 2, 0, 0, 0, 2, 0],
|
|
[0, 2, 0, 0, 0, 2, 0],
|
|
[0, 2, 0, 1, 0, 2, 0],
|
|
[0, 2, 0, 0, 0, 2, 0],
|
|
[0, 2, 0, 0, 0, 2, 0],
|
|
[0, 0, 0, 0, 0, 0, 0]]
|
|
|
|
def imageResultHr3(self):
|
|
return [[2, 0, 0, 0, 0, 0, 2],
|
|
[2, 0, 0, 0, 0, 0, 2],
|
|
[2, 0, 0, 0, 0, 0, 2],
|
|
[2, 0, 0, 1, 0, 0, 2],
|
|
[2, 0, 0, 0, 0, 0, 2],
|
|
[2, 0, 0, 0, 0, 0, 2],
|
|
[2, 0, 0, 0, 0, 0, 2]]
|
|
|
|
def imageResultHOut(self):
|
|
return [[0, 2, 0, 0, 0, 2, 0],
|
|
[0, 2, 0, 1, 0, 2, 0],
|
|
[0, 2, 0, 0, 0, 2, 0]]
|
|
|
|
def imageResultMultiH(self):
|
|
return [[0, 0, 2, 2, 0, 0, 2],
|
|
[0, 2, 2, 2, 0, 2, 2],
|
|
[0, 1, 2, 2, 1, 2, 2],
|
|
[0, 2, 2, 1, 0, 2, 2],
|
|
[0, 2, 2, 2, 0, 2, 2],
|
|
[0, 2, 2, 0, 0, 2, 2],
|
|
[0, 0, 2, 0, 1, 0, 2]]
|
|
|
|
|
|
#vertical results
|
|
def imageResultVr2(self):
|
|
return [[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 2, 2, 2, 2, 2, 0],
|
|
[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 0, 0, 1, 0, 0, 0],
|
|
[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 2, 2, 2, 2, 2, 0],
|
|
[0, 0, 0, 0, 0, 0, 0]]
|
|
|
|
def imageResultVr3(self):
|
|
return [[2, 2, 2, 2, 2, 2, 2],
|
|
[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 0, 0, 1, 0, 0, 0],
|
|
[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 0, 0, 0, 0, 0, 0],
|
|
[2, 2, 2, 2, 2, 2, 2]]
|
|
|
|
def imageResultVOut(self):
|
|
return [[0, 0, 0],
|
|
[2, 2, 2],
|
|
[0, 0, 0],
|
|
[0, 1, 0],
|
|
[0, 0, 0],
|
|
[2, 2, 2],
|
|
[0, 0, 0]]
|
|
|
|
def imageResultMultiV(self):
|
|
return [[2, 2, 2, 2, 2, 2, 2],
|
|
[0, 2, 2, 2, 2, 2, 0],
|
|
[0, 1, 0, 0, 1, 0, 0],
|
|
[0, 0, 0, 1, 0, 0, 0],
|
|
[2, 2, 2, 2, 2, 2, 2],
|
|
[0, 2, 2, 2, 2, 2, 0],
|
|
[0, 0, 0, 0, 1, 0, 0]]
|
|
|
|
#two results
|
|
|
|
def imageResult2(self):
|
|
return [[0, 0, 0, 0, 0, 0, 0],
|
|
[0, 2, 2, 2, 2, 2, 0],
|
|
[0, 2, 0, 0, 0, 2, 0],
|
|
[0, 2, 0, 1, 0, 2, 0],
|
|
[0, 2, 0, 0, 0, 2, 0],
|
|
[0, 2, 2, 2, 2, 2, 0],
|
|
[0, 0, 0, 0, 0, 0, 0]]
|
|
|
|
def imageResultMulti2(self):
|
|
return [[2, 2, 2, 2, 2, 2, 2],
|
|
[0, 0, 0, 0, 0, 0, 2],
|
|
[0, 1, 0, 0, 1, 0, 2],
|
|
[0, 0, 0, 1, 0, 0, 2],
|
|
[2, 2, 0, 0, 0, 2, 2],
|
|
[2, 2, 2, 0, 0, 0, 2],
|
|
[2, 2, 2, 0, 1, 0, 2]]
|
|
|
|
|
|
|
|
def test_hori(self):
|
|
|
|
self.assertEqual(scanHorizontal(self.imageTest(), 2), self.imageResultHr2())
|
|
self.assertEqual(scanHorizontal(self.imageTest(), 3), self.imageResultHr3())
|
|
self.assertEqual(scanHorizontal(self.imageTestHOut(), 2), self.imageResultHOut())
|
|
self.assertEqual(scanHorizontal(self.imageTest(), 4), self.imageTest())
|
|
self.assertEqual(scanHorizontal(self.imageTestMulti(), 2), self.imageResultMultiH())
|
|
|
|
def test_vert(self):
|
|
|
|
self.assertEqual(scanVertical(self.imageTest(), 2), self.imageResultVr2())
|
|
self.assertEqual(scanVertical(self.imageTest(), 3), self.imageResultVr3())
|
|
self.assertEqual(scanVertical(self.imageTestVOut(), 2), self.imageResultVOut())
|
|
self.assertEqual(scanVertical(self.imageTest(), 4), self.imageTest())
|
|
self.assertEqual(scanVertical(self.imageTestMulti(), 2), self.imageResultMultiV())
|
|
|
|
def test_two(self):
|
|
|
|
self.assertEqual(twoRemoving(self.imageTest2(), 2), self.imageResult2())
|
|
self.assertEqual(twoRemoving(self.imageTest2(), 3), self.imageTest())
|
|
self.assertEqual(twoRemoving(self.imageTest2(), 7), self.imageTest())
|
|
self.assertEqual(twoRemoving(self.imageTestMulti2(), 2), self.imageResultMulti2())
|