nama_test

nama_test.py

from unittest import TestCase, main


def nama_req(val):
    return 'Nama Team' if val % 35 == 0 \
        else 'Nama' if val % 5 == 0 \
        else 'Team' if val % 7 == 0 \
        else val


class Nama_tests(TestCase):
    vals = [nama_req(x) for x in range(1, 101)]

    def test_5_mul(self):
        """Testar os multiplos de 5."""
        mul_5 = range(0, 101, 5)

        for val in mul_5:
            self.assertNotIn(val, self.vals)

    def test_7_mul(self):
        """Testar os multiplos de 7."""
        mul_7 = range(0, 101, 7)

        for val in mul_7:
            self.assertNotIn(val, self.vals)

    def test_35_mul(self):
        """Testar os multiplos de 35 (35, 70)."""
        mul_35 = range(0, 101, 35)

        for val in mul_35:
            self.assertNotIn(val, self.vals)

    def test_Nama(self):
        """
        Testa quantidade de multiplos de 5 com Nama.

        Foram subtraidos 3 do valor total pois também são multiplos de 35
            e foram iniciados em zero
        """
        quant_Nama = len(range(0, 101, 5)) - 3

        self.assertEqual(quant_Nama, self.vals.count('Nama'))

    def test_Team(self):
        """
        Testa quantidade de multiplos de 7 com Team.

        Foram subtraidos 3 do valor total pois também são multiplos de 35
            e foram iniciados em zero
        """
        quant_Team = len(range(0, 101, 7)) - 3

        self.assertEqual(quant_Team, self.vals.count('Team'))

    def test_Nama_Team(self):
        """
        Testa quantidade de multiplos de 35 com Nama Team.

        len - 1 pois inicia em 0
        """
        quant_Nama_Team = len(range(0, 101, 35)) - 1

        self.assertEqual(quant_Nama_Team, self.vals.count('Nama Team'))

    def test_Nama_Team_pos(self):
        pos = [x for x in range(0, 101) if x % 35 == 0][1:]

        for el in pos:
            self.assertEqual(self.vals[el-1], 'Nama Team')

if __name__ == '__main__':
    main()