NameError: имя 'zombie_array' не определено

Question

У меня есть код Python (q4.py), указанный ниже:

#!/usr/bin/env python


def DFS(j, visited, zombies, row):
    for k in range(row):
        if zombies[j][k] == '1' and visited[j][k] == False and visited[k][j] == False:
            visited[j][k] = True
            visited[k][j] = True
            DFS(k, visited, zombies, row)


def zombieCluster(zombies):
    row = len(zombies)
    col = len(zombies[0])
    count = 0
    if row == 0 or col == 0:
        return count
    visited = [[False for j in range(col)] for i in range(row)]
    for i in range(row):
        bol = False
        for j in range(row):
            if zombies[i][j] == '1' and visited[i][j] == False and visited[j][i] == False:
                visited[i][j] = True
                visited[j][i] = True
                DFS(j, visited, zombies, row)
                if bol == 0:
                    count += 1
                    bol = True
    return count


if __name__ == '__main__':
    # array input of zombie
    zombie_array = list()
    zombie_count = int(input())
    for i in range(int(zombie_count)):
        n = raw_input()
        zombie_array.append(str(n))

# zombie_array = ["1100" ,"1110", "0110", "0001"]
# print out the result
print(zombieCluster(zombie_array))

Я пишу тестовый код (test_q4.py) для этого кода, приведенного ниже:

# -*- coding: utf-8 -*-
import unittest
import os
import sys

BASEDIR = os.path.abspath(os.path.join(os.path.dirname(os.path.abspath(__file__)), ".."))
sys.path.insert(0, BASEDIR)

from q4 import zombieCluster


class Test(unittest.TestCase):
    def testzombieCluster(self):
        zombie_array = ["1100", "1110", "0110", "0001"]
        expect = 2
        result = zombieCluster(zombie_array)
        self.assertEqual(result, expect)


if __name__ == '__main__':
     unittest.main()

Но когда я запускаю эту команду python test_q4.py из консоли, она выдает ошибку:

Traceback (most recent call last):
  File "test_q4.py", line 9, in <module>
    from q4 import zombieCluster
  File "/home/rowle/Desktop/python/solution/q4.py", line 42, in <module>
    print(zombieCluster(zombie_array))
NameError: name 'zombie_array' is not defined

Когда я удалил print(zombieCluster(zombie_array)) из исходного кода, мой тест выполняется нормально, но если я удалю эту строку,Я не увижу никаких выходных данных.

В этих условиях Что мне делать в тестовом коде?Кроме того, как я могу проверить main функцию исходного кода?

Answer 1

Создайте def main () и запишите код следующим образом:

def main():
    # array input of zombie
    zombie_array = list()
    zombie_count = int(input())
    for i in range(int(zombie_count)):
        n = raw_input()
        zombie_array.append(str(n))

    # zombie_array = ["1100" ,"1110", "0110", "0001"]
    # print out the result
    print(zombieCluster(zombie_array))


if __name__ == '__main__':
    main()