diff --git a/jest.config.js b/jest.config.js
index 837b2d397..09c257006 100644
--- a/jest.config.js
+++ b/jest.config.js
@@ -2,7 +2,7 @@ module.exports = {
   testEnvironment: 'node',
   clearMocks: true,
   setupFilesAfterEnv: ["jest-extended/all"],
-  roots: ["<rootDir>/tests/jest"],
+  roots: ["<rootDir>/tests/jest", "<rootDir>/src"],
   transform: {
     '^.+\\.tsx?$': ['ts-jest', {
       ...require('./tsconfig.test.json')
diff --git a/src/util/addClass.test.ts b/src/util/addClass.test.ts
new file mode 100644
index 000000000..5bccb7fa1
--- /dev/null
+++ b/src/util/addClass.test.ts
@@ -0,0 +1,84 @@
+import { addClass, setClass } from "./className";
+
+describe("className", () => {
+  describe("setClass", () => {
+    it("should set class name to an element", () => {
+      const el = document.createElement("div");
+      el.className = "firstClass";
+
+      setClass(el, "secondClass");
+
+      expect(el.className).toBe("secondClass");
+    });
+
+    it("should set class name to an SVG element", () => {
+      const el = document.createElementNS("http://www.w3.org/2000/svg", "svg");
+      el.setAttribute("class", "firstClass");
+
+      setClass(el, "secondClass");
+
+      expect(el.getAttribute("class")).toBe("secondClass");
+    });
+  });
+
+  describe("addClass", () => {
+    test("should append when className is empty", () => {
+      const el = document.createElement("div");
+      addClass(el, "myClass");
+      expect(el.className).toBe("myClass");
+    });
+
+    test("should append when className is NOT empty", () => {
+      const el = document.createElement("div");
+      el.className = "firstClass";
+
+      addClass(el, "secondClass");
+
+      expect(el.className).toBe("firstClass secondClass");
+    });
+
+    test("should not append duplicate classNames to elements", () => {
+      const el = document.createElement("div");
+      el.className = "firstClass";
+
+      addClass(el, "firstClass");
+
+      expect(el.className).toBe("firstClass");
+    });
+
+    test("should not append duplicate list of classNames to elements", () => {
+      const el = document.createElement("div");
+      el.className = "firstClass firstClass";
+
+      addClass(el, "firstClass", "firstClass", "firstClass");
+
+      expect(el.className).toBe("firstClass");
+    });
+
+    test("should not append duplicate list of classNames to an empty className", () => {
+      const el = document.createElement("div");
+
+      addClass(el, "firstClass", "firstClass", "firstClass");
+
+      expect(el.className).toBe("firstClass");
+    });
+
+    test("should append lassNames to an SVG", () => {
+      const el = document.createElementNS("http://www.w3.org/2000/svg", "svg");
+      el.setAttribute("class", "firstClass");
+
+      addClass(el, "secondClass", "thirdClass");
+
+      expect(el.getAttribute('class')).toBe("firstClass secondClass thirdClass");
+    });
+
+    test("should not append duplicate list of classNames to an empty className of SVG", () => {
+      const el = document.createElementNS("http://www.w3.org/2000/svg", "svg");
+      el.setAttribute("class", "firstClass");
+
+      addClass(el, "firstClass", "firstClass", "firstClass");
+
+      expect(el.getAttribute('class')).toBe("firstClass");
+    });
+  });
+});
diff --git a/src/util/className.ts b/src/util/className.ts
index 22f9c03be..e44eb6100 100644
--- a/src/util/className.ts
+++ b/src/util/className.ts
@@ -2,7 +2,10 @@
  * Append CSS classes to an element
  * @api private
  */
-export const addClass = (element: HTMLElement, ...classNames: string[]) => {
+export const addClass = (
+  element: HTMLElement | SVGElement,
+  ...classNames: string[]
+) => {
   for (const className of classNames) {
     if (element instanceof SVGElement) {
       // svg
@@ -24,7 +27,15 @@ export const addClass = (element: HTMLElement, ...classNames: string[]) => {
   }
 };
 
-export const setClass = (element: HTMLElement, ...classNames: string[]) => {
+/**
+ * Set CSS classes to an element
+ * @param element element to set class
+ * @param classNames list of class names
+ */
+export const setClass = (
+  element: HTMLElement | SVGElement,
+  ...classNames: string[]
+) => {
   const className = classNames.filter(Boolean).join(" ");
 
   if (element instanceof SVGElement) {
diff --git a/tests/jest/util/addClass.test.ts b/tests/jest/util/addClass.test.ts
deleted file mode 100644
index d99902ae3..000000000
--- a/tests/jest/util/addClass.test.ts
+++ /dev/null
@@ -1,44 +0,0 @@
-import addClass from "../../../src/util/className";
-
-describe("addClass", () => {
-  test("should append when className is empty", () => {
-    const el = document.createElement("div");
-    addClass(el, "myClass");
-    expect(el.className).toBe("myClass");
-  });
-
-  test("should append when className is NOT empty", () => {
-    const el = document.createElement("div");
-    el.className = "firstClass";
-
-    addClass(el, "secondClass");
-
-    expect(el.className).toBe("firstClass secondClass");
-  });
-
-  test("should not append duplicate classNames to elements", () => {
-    const el = document.createElement("div");
-    el.className = "firstClass";
-
-    addClass(el, "firstClass");
-
-    expect(el.className).toBe("firstClass");
-  });
-
-  test("should not append duplicate list of classNames to elements", () => {
-    const el = document.createElement("div");
-    el.className = "firstClass firstClass";
-
-    addClass(el, "firstClass firstClass firstClass");
-
-    expect(el.className).toBe("firstClass");
-  });
-
-  test("should not append duplicate list of classNames to an empty className", () => {
-    const el = document.createElement("div");
-
-    addClass(el, "firstClass firstClass firstClass");
-
-    expect(el.className).toBe("firstClass");
-  });
-});