made some checks in TextClassifier

textclassifier/TextClassifier.py

@@ -22,6 +22,14 @@ class TextClassifier:
         :param to_date: string of the format 'YYYY-MM-DD'.
         :param num_tweets: integer value of the maximum number of tweets to be scraped.
         """
+        # Make sure to_date is later than from_date
+        assert from_date < to_date, "from_date must be earlier than to_date"
+        # Make sure the dates are in the correct format
+        assert re.match(r'^\d{4}-\d{2}-\d{2}$', from_date) is not None, "from_date must be in the format YYYY-MM-DD"
+        # Make sure user_name is not empty
+        assert user_name is not None, "user_name cannot be empty"
+        # Make sure num_tweets is a positive integer
+        assert num_tweets > 0, "num_tweets must be a positive integer"
 
         self.model_name = model_name
         self.from_date = from_date