pyembroidery/test/test_convert_pes.py

177 wiersze
6.7 KiB
Python

from __future__ import print_function
import unittest
from pyembroidery import *
from test.pattern_for_tests import *
class TestConverts(unittest.TestCase):
def position_equals(self, stitches, j, k):
self.assertEqual(stitches[j][:1], stitches[k][:1])
def test_convert_pes_to_u01(self):
file1 = "convert_u01.pes"
file2 = "converted_pes.u01"
write_pes(get_big_pattern(), file1)
f_pattern = read_pes(file1)
write_u01(f_pattern, file2)
t_pattern = read_u01(file2)
self.assertIsNotNone(t_pattern)
self.assertEqual(t_pattern.count_stitch_commands(NEEDLE_SET), 16)
self.assertEqual(t_pattern.count_stitch_commands(STITCH), 16 * 5)
self.position_equals(t_pattern.stitches, 0, -1)
print("pes->u01: ", t_pattern.stitches)
self.addCleanup(os.remove, file1)
self.addCleanup(os.remove, file2)
def test_convert_pes_to_csv(self):
file1 = "convert_csv.pes"
file2 = "converted_pes.csv"
write_pes(get_big_pattern(), file1)
f_pattern = read_pes(file1)
write_csv(f_pattern, file2)
t_pattern = read_csv(file2)
self.assertIsNotNone(t_pattern)
self.assertEqual(t_pattern.count_stitch_commands(COLOR_CHANGE), 15)
self.assertEqual(t_pattern.count_stitch_commands(STITCH), 16 * 5)
self.position_equals(t_pattern.stitches, 0, -1)
print("pes->csv: ", t_pattern.stitches)
self.addCleanup(os.remove, file1)
self.addCleanup(os.remove, file2)
def test_convert_pes_to_exp(self):
file1 = "convert_exp.pes"
file2 = "converted_pes.exp"
write_pes(get_big_pattern(), file1)
f_pattern = read_pes(file1)
write_exp(f_pattern, file2)
t_pattern = read_exp(file2)
self.assertIsNotNone(t_pattern)
self.assertEqual(t_pattern.count_stitch_commands(COLOR_CHANGE), 15)
self.assertEqual(t_pattern.count_stitch_commands(STITCH), 16 * 5)
self.position_equals(t_pattern.stitches, 0, -1)
print("pes->exp: ", t_pattern.stitches)
self.addCleanup(os.remove, file1)
self.addCleanup(os.remove, file2)
def test_convert_pes_to_pes(self):
file1 = "convert_pes.pes"
file2 = "converted_pes.pes"
write_pes(get_big_pattern(), file1)
f_pattern = read_pes(file1)
write_pes(f_pattern, file2)
t_pattern = read_pes(file2)
self.assertIsNotNone(t_pattern)
self.assertEqual(t_pattern.count_stitch_commands(COLOR_CHANGE), 15)
self.assertEqual(t_pattern.count_stitch_commands(STITCH), 16 * 5)
self.position_equals(t_pattern.stitches, 0, -1)
print("pes->pes: ", t_pattern.stitches)
self.addCleanup(os.remove, file1)
self.addCleanup(os.remove, file2)
def test_convert_pes_to_jef(self):
file1 = "convert_jef.pes"
file2 = "converted_pes.jef"
write_pes(get_big_pattern(), file1)
f_pattern = read_pes(file1)
write_jef(f_pattern, file2)
t_pattern = read_jef(file2)
self.assertIsNotNone(t_pattern)
self.assertEqual(t_pattern.count_stitch_commands(COLOR_CHANGE), 15)
self.assertEqual(t_pattern.count_stitch_commands(STITCH), 16 * 5)
self.position_equals(t_pattern.stitches, 0, -1)
print("pes->jef: ", t_pattern.stitches)
self.addCleanup(os.remove, file1)
self.addCleanup(os.remove, file2)
def test_convert_pes_to_pec(self):
file1 = "convert_pec.pes"
file2 = "converted_pes.pec"
write_pes(get_big_pattern(), file1)
f_pattern = read_pes(file1)
write_pec(f_pattern, file2)
t_pattern = read_pec(file2)
self.assertIsNotNone(t_pattern)
self.assertEqual(t_pattern.count_stitch_commands(COLOR_CHANGE), 15)
self.assertEqual(t_pattern.count_stitch_commands(STITCH), 16 * 5)
self.position_equals(t_pattern.stitches, 0, -1)
print("pes->pec: ", t_pattern.stitches)
self.addCleanup(os.remove, file1)
self.addCleanup(os.remove, file2)
def test_convert_pes_to_vp3(self):
file1 = "convert_vp3.pes"
file2 = "converted_pes.vp3"
write_pes(get_big_pattern(), file1)
f_pattern = read_pes(file1)
write_vp3(f_pattern, file2)
t_pattern = read_vp3(file2)
self.assertIsNotNone(t_pattern)
self.assertEqual(t_pattern.count_stitch_commands(COLOR_CHANGE), 15)
self.assertEqual(t_pattern.count_stitch_commands(STITCH), 16 * 5)
self.position_equals(t_pattern.stitches, 0, -1)
print("pes->vp3: ", t_pattern.stitches)
self.addCleanup(os.remove, file1)
self.addCleanup(os.remove, file2)
def test_convert_pes_to_dst(self):
file1 = "convert_dst.pes"
file2 = "converted_pes.dst"
write_pes(get_big_pattern(), file1)
f_pattern = read_pes(file1)
write_dst(f_pattern, file2)
t_pattern = read_dst(file2)
self.assertIsNotNone(t_pattern)
self.assertEqual(t_pattern.count_stitch_commands(COLOR_CHANGE), 15)
self.assertEqual(t_pattern.count_stitch_commands(STITCH), 16 * 5)
self.position_equals(t_pattern.stitches, 0, -1)
print("pes->dst: ", t_pattern.stitches)
self.addCleanup(os.remove, file1)
self.addCleanup(os.remove, file2)
def test_convert_pes_to_gcode(self):
file1 = "convert_gcode.pes"
file2 = "converted_pes.gcode"
write_pes(get_big_pattern(), file1)
f_pattern = read_pes(file1)
write_gcode(f_pattern, file2)
t_pattern = read_gcode(file2)
self.assertIsNotNone(t_pattern)
self.assertEqual(t_pattern.count_stitch_commands(COLOR_CHANGE), 15)
self.assertEqual(t_pattern.count_stitch_commands(STITCH), 16 * 5)
self.position_equals(t_pattern.stitches, 0, -1)
print("pes->gcode: ", t_pattern.stitches)
self.addCleanup(os.remove, file1)
self.addCleanup(os.remove, file2)
def test_convert_pes_to_xxx(self):
file1 = "convert_xxx.pes"
file2 = "converted_pes.xxx"
write_pes(get_big_pattern(), file1)
f_pattern = read_pes(file1)
write_xxx(f_pattern, file2)
t_pattern = read_xxx(file2)
self.assertIsNotNone(t_pattern)
self.assertEqual(t_pattern.count_stitch_commands(COLOR_CHANGE), 15)
self.assertEqual(t_pattern.count_stitch_commands(STITCH), 16 * 5)
self.position_equals(t_pattern.stitches, 0, -1)
print("pes->xxx: ", t_pattern.stitches)
self.addCleanup(os.remove, file1)
self.addCleanup(os.remove, file2)
def test_write_pes_long(self):
file1 = "long.pes"
write_pes(get_long_jump(), file1)
self.addCleanup(os.remove, file1)