Moved code to pgexecute

1 files changed, 10 insertions, 1 deletions

diff --git a/tests/parseutils/test_parseutils.py b/tests/parseutils/test_parseutils.py
index 4ffc7ee3..28918c07 100644
--- a/tests/parseutils/test_parseutils.py
+++ b/tests/parseutils/test_parseutils.py
@@ -1,5 +1,5 @@
 import pytest
-from pgcli.packages.parseutils.tables import extract_tables
+from pgcli.packages.parseutils.tables import extract_tables, schema_table_split
 from pgcli.packages.parseutils.utils import find_prev_keyword, is_open_quote
 
 
@@ -265,3 +265,12 @@ def test_is_open_quote__closed(sql):
 ])
 def test_is_open_quote__open(sql):
     assert is_open_quote(sql)
+
+
+def test_schema_table_split():
+    assert schema_table_split('myschema.mytable') == ('myschema', 'mytable')
+    assert schema_table_split('mytable') == ('%', 'mytable')
+    assert schema_table_split('') == ('%', '')
+    assert schema_table_split('myschema.mytable.foo') == (
+        'myschema', 'mytable.foo')
+    assert schema_table_split('%') == ('%', '%')