From 1487614e86c44aebdc56566faea80ac0007895b1 Mon Sep 17 00:00:00 2001
From: Rostyslav Hnatyshyn <rostyslav.hnatyshyn@gmail.com>
Date: Wed, 3 Jan 2024 12:37:04 -0700
Subject: [PATCH] split ai and entity test

---
 tests/ai_test.rs     | 22 +---------------------
 tests/entity_test.rs | 27 +++++++++++++++++++++++++++
 2 files changed, 28 insertions(+), 21 deletions(-)
 create mode 100644 tests/entity_test.rs

diff --git a/tests/ai_test.rs b/tests/ai_test.rs
index 131e288..87bffcf 100644
--- a/tests/ai_test.rs
+++ b/tests/ai_test.rs
@@ -2,7 +2,7 @@ use antf::lib::screen::init_screen;
 use antf::lib::point::Point;
 use antf::lib::ai::AIGoal;
 use antf::lib::world::{World, simulate, render};
-use antf::lib::entity::{Entities, Ant, FoodGenerator};
+use antf::lib::entity::{Entities, Ant};
 
 use ncurses::*;
 use std::thread::sleep;
@@ -34,23 +34,3 @@ fn test_reach_astar() {
     clear();
     endwin();
 }
-
-#[test]
-fn test_foodgen() {
-    let mut board = init_screen(); 
-    let mut world = World::new();
-
-    let mut entities = Entities::new();
-    let fg = FoodGenerator::new();
-    entities.add_entity(&fg);
-
-    for _ in 0..60 {
-        // TODO: add way to break out of the loop by hitting a random key
-        simulate(&mut entities, &mut world, &mut board);
-        render(&entities, &world, &board);
-        sleep(time::Duration::from_millis(100));
-        refresh();
-    }
-    clear();
-    endwin();
-}
diff --git a/tests/entity_test.rs b/tests/entity_test.rs
new file mode 100644
index 0000000..6053ed2
--- /dev/null
+++ b/tests/entity_test.rs
@@ -0,0 +1,27 @@
+use antf::lib::screen::init_screen;
+use antf::lib::world::{World, simulate, render};
+use antf::lib::entity::{Entities, FoodGenerator};
+
+use ncurses::*;
+use std::thread::sleep;
+use std::time;
+
+#[test]
+fn test_foodgen() {
+    let mut board = init_screen(); 
+    let mut world = World::new();
+
+    let mut entities = Entities::new();
+    let fg = FoodGenerator::new();
+    entities.add_entity(&fg);
+
+    for _ in 0..60 {
+        // TODO: add way to break out of the loop by hitting a random key
+        simulate(&mut entities, &mut world, &mut board);
+        render(&entities, &world, &board);
+        sleep(time::Duration::from_millis(100));
+        refresh();
+    }
+    clear();
+    endwin();
+}