Add tests, new class for FA, remove .rst

Author: firefly-cpp

FireflyAlgorithm.py

@@ -0,0 +1,12 @@
+import numpy as np
+from fireflyalgorithm import FireflyAlgorithm
+
+
+def sphere(x):
+    return np.sum(x ** 2)
+
+
+FA = FireflyAlgorithm(function=sphere, dim=10, lb=-5, ub=5, max_evals=10000)
+best = FA.move()
+
+print(best)

README.rst

@@ -0,0 +1,5 @@
+from fireflyalgorithm.fireflyalgorithm import FireflyAlgorithm
+
+__all__ = ['FireflyAlgorithm']
+
+__version__ = '0.2'

examples/run.py

@@ -0,0 +1,41 @@
+import numpy as np
+from numpy.random import default_rng
+
+
+class FireflyAlgorithm():
+    def __init__(self, function, dim, lb, ub, max_evals, pop_size=20, alpha=1.0, betamin=1.0, gamma=0.01, seed=None):
+        self.function = function
+        self.dim = dim
+        self.lb = lb
+        self.ub = ub
+        self.max_evals = max_evals
+        self.pop_size = pop_size
+        self.alpha = alpha
+        self.betamin = betamin
+        self.gamma = gamma
+        self.seed = seed
+
+    def move(self):
+        rng = default_rng(self.seed)
+        fireflies = rng.uniform(self.lb, self.ub, (self.pop_size, self.dim))
+        intensity = np.apply_along_axis(self.function, 1, fireflies)
+        best = np.min(intensity)
+
+        evaluations = self.pop_size
+        new_alpha = self.alpha
+        search_range = self.ub - self.lb
+
+        while evaluations <= self.max_evals:
+            new_alpha *= 0.97
+            for i in range(self.pop_size):
+                for j in range(self.pop_size):
+                    if intensity[i] >= intensity[j]:
+                        r = np.sum(np.square(fireflies[i] - fireflies[j]), axis=-1)
+                        beta = self.betamin * np.exp(-self.gamma * r)
+                        steps = new_alpha * (rng.random(self.dim) - 0.5) * search_range
+                        fireflies[i] += beta * (fireflies[j] - fireflies[i]) + steps
+                        fireflies[i] = np.clip(fireflies[i], self.lb, self.ub)
+                        intensity[i] = self.function(fireflies[i])
+                        evaluations += 1
+                        best = min(intensity[i], best)
+        return best

fireflyalgorithm/__init__.py

@@ -0,0 +1 @@
+ 

fireflyalgorithm/fireflyalgorithm.py

@@ -0,0 +1,6 @@
+import logging
+
+
+def pytest_configure(config):
+    r"""Disable verbose output when running tests."""
+    logging.basicConfig(level=logging.DEBUG)

fireflyalgorithm/tests/__init__.py

@@ -0,0 +1,14 @@
+from unittest import TestCase
+
+import numpy as np
+from fireflyalgorithm import FireflyAlgorithm
+import os
+
+def sphere(x):
+    return np.sum(x ** 2)
+
+class TestFA(TestCase):
+    def test_algorithm(self):
+        FA = FireflyAlgorithm(function=sphere, dim=10, lb=-5, ub=5, max_evals=10000)
+        best = FA.move()
+        self.assertLess(best, 5)

fireflyalgorithm/tests/conftest.py

@@ -6,17 +6,21 @@ authors = ["firefly-cpp"]
 license = "MIT"
 readme = "README.md"
 
-packages = [
-    {include = "FireflyAlgorithm.py"},
+include = [
+    { path="LICENSE", format="sdist" }
 ]
 
+
 [tool.poetry.dependencies]
 python = "^3.7"
 numpy = [
     { version = "^1.21.5", python = ">=3.7,<3.11" },
     { version = "^1.22.0", python = "^3.11" }
 ]
 
+[tool.poetry.dev-dependencies]
+pytest = "^7.0.1"
+
 [build-system]
 requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"

fireflyalgorithm/tests/test_firefly.py

@@ -0,0 +1 @@
+ 

pyproject.toml

@@ -0,0 +1,3 @@
+from fireflyalgorithm.tests.conftest import pytest_configure
+
+__all__ = ["pytest_configure"]