Merge pull request #19 from jomonson/patch-1

Treat empty tables the same as non-empty tables

CHANGES.rst

@@ -12,7 +12,7 @@ Removed
 
 Fixed
 -----
-
+- Empty tables are now handled correctly.
 - Very large tables are now sorted without crashing. This is done by merge sorting
   in temporary files.
 
@@ -37,7 +37,7 @@ Added
 Added
 -----
 
-Document ``pg_incremental_backup.py`` in the README file
+- Document ``pg_incremental_backup.py`` in the README file
 
 
 0.9_ / 2015-03-10

pgtricks/pg_dump_splitsort.py

@@ -11,7 +11,7 @@
 
 from pgtricks.mergesort import MergeSort
 
-COPY_RE = re.compile(r'COPY .*? \(.*?\) FROM stdin;\n$')
+COPY_RE = re.compile(r"COPY\s+\S+\s+(\(.*?\)\s+)?FROM\s+stdin;\n$")
 KIBIBYTE, MEBIBYTE, GIBIBYTE = 2**10, 2**20, 2**30
 MEMORY_UNITS = {"": 1, "k": KIBIBYTE, "m": MEBIBYTE, "g": GIBIBYTE}
 

pgtricks/tests/test_pg_dump_splitsort.py

@@ -3,7 +3,34 @@
 
 import pytest
 
-from pgtricks.pg_dump_splitsort import linecomp, memory_size, split_sql_file, try_float
+from pgtricks.pg_dump_splitsort import (
+    COPY_RE,
+    linecomp,
+    memory_size,
+    split_sql_file,
+    try_float,
+)
+
+
+@pytest.mark.parametrize(
+    ("test_input", "expected"),
+    [
+        ("COPY table_name (column1, column2) FROM stdin;\n", True),
+        ("COPY   table_name   (column1,   column2)   FROM   stdin;\n", True),
+        ("COPY table_name FROM stdin;\n", True),
+        ("COPY   table_name   FROM   stdin;\n", True),
+        ("COPYtable_name FROM stdin;\n", False),  # No space after COPY
+        ("COPY table_name FROMstdin;\n", False),  # No space before stdin
+        ("COPY table_name FROM ;\n", False),  # Missing stdin
+        ("COPY table_name stdin;\n", False),  # Missing FROM
+        ("COPY FROM stdin;\n", False),  # Missing table name
+    ],
+)
+def test_sql_copy_regular_expression(test_input, expected):
+    """Test that `COPY_RE` matches/doesn't match the expected strings."""
+    result = COPY_RE.match(test_input) is not None
+
+    assert result == expected
 
 
 @pytest.mark.parametrize(