From c471dbf19e0dec2f6d7829b12ac32ed7b589ca1b Mon Sep 17 00:00:00 2001
From: MstrPikachu <31784486+MstrPikachu@users.noreply.github.com>
Date: Mon, 13 Dec 2021 01:35:37 -0500
Subject: [PATCH] Add representation invariants to visualization.py

---
 src/visualization.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/visualization.py b/src/visualization.py
index 4da267a..426f48c 100644
--- a/src/visualization.py
+++ b/src/visualization.py
@@ -22,12 +22,24 @@ class UserFloat:
 
     This is used for both COVID tweet frequency and popularity ratio data, because both of these
     are floating point data.
+
+    Representation Invariants:
+        - self.name != ''
+
     """
     name: str
     data: float
 
 
 class Sample:
+    """
+    A sample of many users, containing statistical data that will be used in graphs.
+
+    Representation Invariants:
+        - self.name != ''
+        - all(name != '' for name in self.users)
+
+    """
     name: str
     users: list[str]
     # Total frequencies of all posts for each user across all dates (sorted)