def test_count_1_object_irregular(self): a = np.array([[0., 0., 0., 0., 0.], [0., 1., 1., 1., 0.], [0., 1., 1., 1., 1.], [0., 1., 1., 1., 0.], [0., 0., 1., 0., 0.], [0., 0., 0., 0., 0.], [0., 0., 0., 0., 0.], [0., 0., 0., 0., 0.], [0., 0., 0., 0., 0.], [0., 0., 0., 0., 0.]]) assert_equal(count_objects(a), 1)
def test_count_1_object_rectangle(self): a = np.array([[0., 0., 0., 0., 0.], [0., 1., 1., 1., 0.], [0., 1., 1., 1., 1.], [0., 1., 1., 1., 0.], [0., 0., 1., 0., 0.], [0., 0., 0., 0., 0.], [0., 0., 0., 0., 0.], [0., 0., 0., 0., 0.], [0., 0., 0., 0., 0.], [0., 0., 0., 0., 0.]]) assert_equal(count_objects(a), 1)
def test_count_objects_2_objects(self): a = np.array([[0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], [0., 1., 1., 1., 0., 0., 0., 0., 0., 0.], [0., 1., 1., 1., 0., 0., 0., 0., 0., 0.], [0., 1., 1., 1., 0., 0., 0., 0., 0., 0.], [0., 0., 0., 0., 0., 1., 1., 0., 0., 0.], [0., 0., 0., 0., 0., 1., 1., 0., 0., 0.], [0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], [0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], [0., 0., 0., 0., 0., 0., 0., 0., 0., 0.], [0., 0., 0., 0., 0., 0., 0., 0., 0., 0.]]) assert_equal(count_objects(a), 2)
def test_count_objects(self): a = np.zeros([6, 6], dtype=np.float64) assert_equal(count_objects(a), 0)
def test_count_all_ones(self): a = np.ones([6, 6], dtype=np.int32) assert_equal(count_objects(a), 1)
def test_count_1_object(self): a = np.zeros([6, 6], dtype=np.float64) a[1:-1, 1:-1] = 1 assert_equal(count_objects(a), 1)