import pytest

from examples.gbest_pso import main as gbest
from examples.lbest_pso import main as lbest
from examples.gc_pso import main as gc
from examples.pso_optimizer import main as pso_optimizer


@pytest.mark.parametrize("dimension", [
    1,
    30
])
@pytest.mark.parametrize("iterations", [
    3
])
def test_gbest_pso(dimension, iterations):
    gbest(dimension, iterations)


@pytest.mark.parametrize("dimension", [
    1,
    30
])
@pytest.mark.parametrize("iterations", [
    3
])
def test_lbest_pso(dimension, iterations):
    lbest(dimension, iterations)


@pytest.mark.parametrize("dimension", [
    1,
    30
])
@pytest.mark.parametrize("iterations", [
    3
])
def test_gc_pso(dimension, iterations):
    gc(dimension, iterations)


@pytest.mark.parametrize("dimension", [
    1,
    30
])
@pytest.mark.parametrize("iterations", [
    3
])
def test_gc_pso(dimension, iterations):
    pso_optimizer(dimension, iterations)

@@ -1,63 +1,634 @@
-import pytest%0A%0Afrom examples.gbest_pso import main as gbest
+# Copyright 2016 Andrich van Wyk%0A#%0A# Licensed under the Apache License, Version 2.0 (the %22License%22);%0A# you may not use this file except in compliance with the License.%0A# You may obtain a copy of the License at%0A#%0A# http://www.apache.org/licenses/LICENSE-2.0%0A#%0A# Unless required by applicable law or agreed to in writing, software%0A# distributed under the License is distributed on an %22AS IS%22 BASIS,%0A# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.%0A# See the License for the specific language governing permissions and%0A# limitations under the License.%0A%0A%22%22%22 Tests ensuring examples execute%0A%22%22%22%0Aimport pytest%0A
 %0Afro
@@ -638,17 +638,17 @@
 xamples.
-l
+g
 best_pso
@@ -663,17 +663,17 @@
 main as 
-l
+g
 best%0Afro
@@ -708,16 +708,61 @@
 n as gc%0A
+from examples.lbest_pso import main as lbest%0A
 from exa

Add license header and file documentation