Cleaned up test files to only be source files

1 files changed, 16 insertions, 2 deletions

diff --git a/sigcore/test/TestHat.cpp b/sigcore/test/TestHat.cpp
index 215744a7..78d39f25 100644
--- a/sigcore/test/TestHat.cpp
+++ b/sigcore/test/TestHat.cpp
@@ -15,12 +15,26 @@
  * with this program.  If not, see <http://www.gnu.org/licenses/>.
  */
 
-// Header include
-#include "TestHat.h"
+// Sigcore includes
+#include "../Hat.h"
 
 // Qt includes
 #include <QtTest/QTest>
 
+class TestHat : public QObject
+{
+    Q_OBJECT
+    
+    private slots:
+        void initTestCase();
+        void pick();
+        void take();
+        void takeAndClear();
+        void setCount();
+        void add();
+        void chance();
+};
+
 void TestHat::initTestCase()
 {
     qsrand(QDateTime::currentDateTime().toTime_t());