Function test_array_generation

benchmarks/test_array_benchmark_simple.py:41–74

Test array data generation.

()

            os.remove(temp_db)

def test_array_generation():
    """Test array data generation."""
    print("\n🧪 Testing array data generation...")

    runner = ArrayBenchmarkRunner(iterations=3, port=15505)

    try:
        # Test different array sizes
        for size in [5, 10, 20]:
            arrays = runner.generate_test_arrays(size)

            # Verify all array types are present
            expected_keys = {"int_array", "bigint_array", "text_array", "float_array", "bool_array"}
            if not expected_keys.issubset(arrays.keys()):
                raise ValueError(f"Missing array types: {expected_keys - arrays.keys()}")

            # Verify array sizes
            for key, array in arrays.items():
                if len(array) != size:
                    raise ValueError(f"{key} has wrong size: {len(array)} != {size}")

            # Verify JSON serialization works
            for key, array in arrays.items():
                json_str = json.dumps(array)
                restored = json.loads(json_str)
                if restored != array:
                    raise ValueError(f"{key} JSON serialization failed")

        print("✅ Array generation test passed!")
        return True

    except Exception as e:
        print(f"❌ Array generation test failed: {e}")
        return False

def main():
    """Run all tests."""