update 791

Author: fishercoder1534

Diff for: src/main/java/com/fishercoder/solutions/_791.java

@@ -5,13 +5,13 @@
 
 public class _791 {
     public static class Solution1 {
-        public String customSortString(String S, String T) {
+        public String customSortString(String order, String s) {
             Map<Character, Integer> map = new HashMap<>();
-            for (char c : T.toCharArray()) {
+            for (char c : s.toCharArray()) {
                 map.put(c, map.getOrDefault(c, 0) + 1);
             }
             StringBuilder sb = new StringBuilder();
-            for (char c : S.toCharArray()) {
+            for (char c : order.toCharArray()) {
                 if (map.containsKey(c)) {
                     int count = map.get(c);
                     while (count-- > 0) {

Diff for: src/test/java/com/fishercoder/_791Test.java

@@ -1,21 +1,21 @@
 package com.fishercoder;
 
 import com.fishercoder.solutions._791;
-import org.junit.BeforeClass;
-import org.junit.Test;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
 
-import static org.junit.Assert.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertEquals;
 
 public class _791Test {
-  private static _791.Solution1 solution1;
+    private static _791.Solution1 solution1;
 
-  @BeforeClass
-  public static void setup() {
-    solution1 = new _791.Solution1();
-  }
+    @BeforeEach
+    public void setup() {
+        solution1 = new _791.Solution1();
+    }
 
-  @Test
-  public void test1() {
-    assertEquals("cbad", solution1.customSortString("cba", "abcd"));
-  }
+    @Test
+    public void test1() {
+        assertEquals("cbad", solution1.customSortString("cba", "abcd"));
+    }
 }