Messenger_1Million_Stats/test/model/message_test.py

import million.analyze.message_evaluation as msg_val
from test.TestCase import TestCase


class MessageTest(TestCase):

    def test_single_digit(self, overrides=None, exclude=None):
        message = self._message_with_text("1")

        assert 1 == msg_val.get(message)

    def test_message_nominal(self, overrides=None, exclude=None):
        message = self._message_with_text("1234")

        assert 1234 == msg_val.get(message)

    def test_message_with_text(self, overrides=None, exclude=None):
        message = self._message_with_text("... 😏😏 269")

        assert 269 == msg_val.get(message)

    def test_message_with_text_2(self, overrides=None, exclude=None):
        message = self._message_with_text("331 allez la")

        assert 331 == msg_val.get(message)

    def test_message_with_text_3(self, overrides=None, exclude=None):
        message = self._message_with_text("Ok 2160")

        assert 2160 == msg_val.get(message)

    def test_message_value_cut(self, overrides=None, exclude=None):
        message = self._message_with_text("66...😏😏😏9")

        assert 669 == msg_val.get(message)

    def test_message_value_cut_2(self, overrides=None, exclude=None):
        message = self._message_with_text("82heyyyyyy69")

        assert 8269 == msg_val.get(message)

    def test_message_value_cut_2(self, overrides=None, exclude=None):
        message = self._message_with_text("9339 9339 9339 9339")

        assert 9339 == msg_val.get(message)

    def test_message_in_middle(self, overrides=None, exclude=None):
        message = self._message_with_text("A peine 5565 ouais...")

        assert 5565 == msg_val.get(message)