def test_accept(self): vertices = [Point(0, 0, 0), Point(0, 1, 0), Point(1, 0, 0)] building = Building(Point(0, 0, 0), vertices) generator_mock = mock.Mock() calls = [mock.call.start_building(building), mock.call.end_building(building)] building.accept(generator_mock) generator_mock.assert_has_calls(calls)