refactor: move persistence tests to dedicated files

- Created separate test files for each persistence implementation:
  * memory_test.py for InMemoryTables tests
  * json_test.py for FileSystemJsonTables tests
  * jsonl_test.py for FileSystemJsonLTables tests
  * extended_test.py for ExtendedTables tests
  * base_test.py for shared FsTablesTest base class

- Cleaned up tables_test.py to only contain core table/column tests
- Fixed circular import issues by removing persistence imports from tables.py
- Updated import statements across the codebase to use persistence module
- All 178 tests continue to pass with improved organization

Author: brunovcosta

abstra_json_sql/apply.py

@@ -2,7 +2,8 @@
 from .infer import infer_expression
 from .parser import parse_expression
 from .lexer import scan
-from .tables import ITablesSnapshot, Table, ExtendedTables, Column
+from .tables import ITablesSnapshot, Table, Column
+from .persistence import ExtendedTables
 from .field_name import field_name, expression_name
 from .ast import (
     Delete,

abstra_json_sql/cli.py

@@ -1,5 +1,6 @@
 from pathlib import Path
-from .tables import FileSystemJsonLTables, Column, ColumnType, Table
+from .tables import Column, ColumnType, Table
+from .persistence import FileSystemJsonLTables
 from .eval import eval_sql
 from argparse import ArgumentParser
 from csv import DictWriter

abstra_json_sql/eval_test.py

@@ -1,6 +1,7 @@
 from unittest import TestCase
 from .eval import eval_sql
-from .tables import InMemoryTables, Table, Column, ColumnType
+from .tables import Table, Column, ColumnType
+from .persistence import InMemoryTables
 
 
 class TestEvalSQL(TestCase):

abstra_json_sql/persistence/base_test.py

@@ -0,0 +1,42 @@
+from unittest import TestCase
+from tempfile import mkdtemp
+from shutil import rmtree
+from pathlib import Path
+from ..tables import Column, ColumnType, Table
+from .json import FileSystemJsonTables
+from .jsonl import FileSystemJsonLTables
+
+
+class FsTablesTest(TestCase):
+    def setUp(self):
+        self.path = Path(mkdtemp())
+        self.path.mkdir(parents=True, exist_ok=True)
+
+        # Create test table using the new UUID-based system
+        test_table = Table(
+            name="test_table",
+            columns=[
+                Column(name="id", type=ColumnType.int),
+                Column(name="name", type=ColumnType.string),
+            ],
+            data=[],
+        )
+
+        # For FileSystemJsonTables
+        tables_json = FileSystemJsonTables(workdir=self.path)
+        tables_json.add_table(test_table)
+
+        # For FileSystemJsonLTables
+        test_table_l = Table(
+            name="test_table",
+            columns=[
+                Column(name="id", type=ColumnType.int),
+                Column(name="name", type=ColumnType.string),
+            ],
+            data=[],
+        )
+        tables_jsonl = FileSystemJsonLTables(workdir=self.path)
+        tables_jsonl.add_table(test_table_l)
+
+    def tearDown(self):
+        rmtree(self.path)

abstra_json_sql/persistence/extended_test.py

@@ -0,0 +1,86 @@
+import pytest
+from ..tables import Column, ColumnType, Table
+from .memory import InMemoryTables
+from .extended import ExtendedTables
+
+
+class TestExtendedTables:
+    @pytest.fixture
+    def base_tables(self):
+        return InMemoryTables(tables=[])
+
+    @pytest.fixture
+    def extra_table(self):
+        return Table(
+            name="extra",
+            columns=[Column(name="id", type=ColumnType.int)],
+            data=[{"id": 1}],
+        )
+
+    def test_extended_tables_creation(self, base_tables, extra_table):
+        extended = ExtendedTables(base_tables, [extra_table])
+        assert extended.snapshot == base_tables
+        assert len(extended.extra_tables) == 1
+
+    def test_get_table_from_base(self, base_tables, extra_table):
+        base_table = Table(
+            name="base", columns=[Column(name="id", type=ColumnType.int)], data=[]
+        )
+        base_tables.add_table(base_table)
+
+        extended = ExtendedTables(base_tables, [extra_table])
+        retrieved = extended.get_table("base")
+        assert retrieved is not None
+        assert retrieved.name == "base"
+
+    def test_get_table_from_extra(self, base_tables, extra_table):
+        extended = ExtendedTables(base_tables, [extra_table])
+        retrieved = extended.get_table("extra")
+        assert retrieved is not None
+        assert retrieved.name == "extra"
+
+    def test_get_nonexistent_table(self, base_tables, extra_table):
+        extended = ExtendedTables(base_tables, [extra_table])
+        assert extended.get_table("nonexistent") is None
+
+    def test_add_table(self, base_tables, extra_table):
+        extended = ExtendedTables(base_tables, [extra_table])
+        new_table = Table(
+            name="new", columns=[Column(name="id", type=ColumnType.int)], data=[]
+        )
+        extended.add_table(new_table)
+
+        assert len(extended.extra_tables) == 2
+        assert extended.get_table("new") is not None
+
+    def test_remove_table(self, base_tables, extra_table):
+        extended = ExtendedTables(base_tables, [extra_table])
+        extended.remove_table("extra")
+
+        assert len(extended.extra_tables) == 0
+        assert extended.get_table("extra") is None
+
+    def test_insert_extra_table(self, base_tables, extra_table):
+        extended = ExtendedTables(base_tables, [extra_table])
+        extended.insert("extra", {"id": 2})
+
+        table = extended.get_table("extra")
+        assert len(table.data) == 2
+
+    def test_update_extra_table(self, base_tables, extra_table):
+        extended = ExtendedTables(base_tables, [extra_table])
+        extended.update("extra", 0, {"id": 999})
+
+        table = extended.get_table("extra")
+        assert table.data[0]["id"] == 999
+
+    def test_delete_extra_table(self, base_tables, extra_table):
+        extended = ExtendedTables(base_tables, [extra_table])
+        extended.delete("extra", [0])
+
+        table = extended.get_table("extra")
+        assert len(table.data) == 0
+
+
+if __name__ == "__main__":
+    pytest.main([__file__, "-v"])