From 93a25d817628dc79deeee363178e29f269a7496a Mon Sep 17 00:00:00 2001
From: HuyQHoang <huyhq53@gmail.com>
Date: Thu, 2 Jun 2022 13:25:49 +0700
Subject: [PATCH] Add test function for Armstrong

---
 .../com/thealgorithms/maths/TestArmstrong.java    | 15 +++++++++++++++
 1 file changed, 15 insertions(+)
 create mode 100644 src/test/java/com/thealgorithms/maths/TestArmstrong.java

diff --git a/src/test/java/com/thealgorithms/maths/TestArmstrong.java b/src/test/java/com/thealgorithms/maths/TestArmstrong.java
new file mode 100644
index 000000000000..afc1ff8a4b61
--- /dev/null
+++ b/src/test/java/com/thealgorithms/maths/TestArmstrong.java
@@ -0,0 +1,15 @@
+package com.thealgorithms.maths;
+
+import org.junit.jupiter.api.Test;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+public class TestArmstrong {
+
+    @Test
+    public void testArmstrong() {
+        Armstrong armstrong = new Armstrong();
+        assertThat(armstrong.isArmstrong(371)).isTrue();
+        assertThat(armstrong.isArmstrong(200)).isFalse();
+    }
+}