Add files via upload

pyembroidery/PyEmbroidery.py

@@ -452,6 +452,16 @@ def read_jef(f, settings=None, pattern=None):
     return read_embroidery(JefReader, f, settings, pattern)
 
 
+def read_u01(f, settings=None, pattern=None):
+    """Reads fileobject as U01 file"""
+    return read_embroidery(U01Reader, f, settings, pattern)
+
+
+def read_csv(f, settings=None, pattern=None):
+    """Reads fileobject as CSV file"""
+    return read_embroidery(CsvReader, f, settings, pattern)
+
+
 def read(filename, settings=None, pattern=None):
     """Reads file, assuming type by extension"""
     extension = get_extension_by_filename(filename)
@@ -515,30 +525,40 @@ def write_dst(pattern, stream, settings=None):
 
 
 def write_pec(pattern, stream, settings=None):
-    """Writes fileobject as DST file"""
+    """Writes fileobject as PEC file"""
     write_embroidery(PecWriter, pattern, stream, settings)
 
 
 def write_pes(pattern, stream, settings=None):
-    """Writes fileobject as DST file"""
+    """Writes fileobject as PES file"""
     write_embroidery(PesWriter, pattern, stream, settings)
 
 
 def write_exp(pattern, stream, settings=None):
-    """Writes fileobject as DST file"""
+    """Writes fileobject as EXP file"""
     write_embroidery(ExpWriter, pattern, stream, settings)
 
 
 def write_vp3(pattern, stream, settings=None):
-    """Writes fileobject as DST file"""
+    """Writes fileobject as Vp3 file"""
     write_embroidery(Vp3Writer, pattern, stream, settings)
 
 
 def write_jef(pattern, stream, settings=None):
-    """Writes fileobject as DST file"""
+    """Writes fileobject as JEF file"""
     write_embroidery(JefWriter, pattern, stream, settings)
 
 
+def write_u01(pattern, stream, settings=None):
+    """Writes fileobject as U01 file"""
+    write_embroidery(U01Writer, pattern, stream, settings)
+
+
+def write_csv(pattern, stream, settings=None):
+    """Writes fileobject as CSV file"""
+    write_embroidery(CsvWriter, pattern, stream, settings)
+
+
 def write_svg(pattern, stream, settings=None):
     """Writes fileobject as DST file"""
     write_embroidery(SvgWriter, pattern, stream, settings)

test/test_embpattern.py

@@ -18,8 +18,67 @@ class TestEmbpattern(unittest.TestCase):
         stitches = pattern.stitches
         assert (stitches[-1][0] == 0)
         assert (stitches[-1][1] == 0)
-        pattern = pattern.get_normalized_pattern()
-        assert (pattern is not None)
-        stitches = pattern.stitches
-        assert (stitches[-1][0] != 0)
-        assert (stitches[-1][1] != 0)
+
+        write_dst(pattern, "file.dst")
+        dst_pattern = read_dst("file.dst")
+        self.assertIsNotNone(dst_pattern)
+        self.assertEqual(dst_pattern.count_stitch_commands(STITCH), 5)
+        self.assertNotEqual(dst_pattern.stitches[-1][0], 0)
+        self.assertNotEqual(dst_pattern.stitches[-1][1], 0)
+        print("dst: ", dst_pattern.stitches)
+
+        write_exp(pattern, "file.exp")
+        exp_pattern = read_exp("file.exp")
+        self.assertIsNotNone(exp_pattern)
+        self.assertEqual(exp_pattern.count_stitch_commands(STITCH), 5)
+        self.assertNotEqual(exp_pattern.stitches[-1][0], 0)
+        self.assertNotEqual(exp_pattern.stitches[-1][1], 0)
+        print("exp: ", exp_pattern.stitches)
+
+        write_vp3(pattern, "file.vp3")
+        vp3_pattern = read_vp3("file.vp3")
+        self.assertIsNotNone(vp3_pattern)
+        self.assertEqual(vp3_pattern.count_stitch_commands(STITCH), 5)
+        self.assertNotEqual(vp3_pattern.stitches[-1][0], 0)
+        self.assertNotEqual(vp3_pattern.stitches[-1][1], 0)
+        print("vp3: ", vp3_pattern.stitches)
+
+        write_jef(pattern, "file.jef")
+        jef_pattern = read_jef("file.jef")
+        self.assertIsNotNone(jef_pattern)
+        self.assertEqual(jef_pattern.count_stitch_commands(STITCH), 5)
+        self.assertNotEqual(jef_pattern.stitches[-1][0], 0)
+        self.assertNotEqual(jef_pattern.stitches[-1][1], 0)
+        print("jef: ", jef_pattern.stitches)
+
+        write_pec(pattern, "file.pec")
+        pec_pattern = read_pec("file.pec")
+        self.assertIsNotNone(pec_pattern)
+        self.assertEqual(pec_pattern.count_stitch_commands(STITCH), 6)
+        self.assertNotEqual(pec_pattern.stitches[-1][0], 0)
+        self.assertNotEqual(pec_pattern.stitches[-1][1], 0)
+        print("pec: ", pec_pattern.stitches)
+
+        write_pes(pattern, "file.pes")
+        pes_pattern = read_pes("file.pes")
+        self.assertIsNotNone(pes_pattern)
+        self.assertEqual(pes_pattern.count_stitch_commands(STITCH), 6)
+        self.assertNotEqual(pes_pattern.stitches[-1][0], 0)
+        self.assertNotEqual(pes_pattern.stitches[-1][1], 0)
+        print("pes: ", pes_pattern.stitches)
+
+        write_u01(pattern, "file.u01")
+        u01_pattern = read_u01("file.u01")
+        self.assertIsNotNone(u01_pattern)
+        self.assertEqual(u01_pattern.count_stitch_commands(STITCH), 5)
+        self.assertNotEqual(u01_pattern.stitches[-1][0], 0)
+        self.assertNotEqual(u01_pattern.stitches[-1][1], 0)
+        print("u01: ", u01_pattern.stitches)
+
+        write_csv(pattern, "file.csv")
+        csv_pattern = read_csv("file.csv")
+        self.assertIsNotNone(csv_pattern)
+        self.assertEqual(csv_pattern.count_stitch_commands(STITCH), 5)
+        self.assertNotEqual(csv_pattern.stitches[-1][0], 0)
+        self.assertNotEqual(csv_pattern.stitches[-1][1], 0)
+        print("csv: ", csv_pattern.stitches)