mirror of
https://github.com/biopython/biopython.git
synced 2025-10-20 21:53:47 +08:00
57056bbfc4
Reverted unwanted changes (mostly arrays in tests, but also some whitespace in doctests). Remaining changes are standardising spacing between module docstring and imports, and lower-case \x<hex> in strings.
88 lines
2.6 KiB
Python
88 lines
2.6 KiB
Python
# Copyright 2001 by Brad Chapman. All rights reserved.
|
|
# This code is part of the Biopython distribution and governed by its
|
|
# license. Please see the LICENSE file that should have been included
|
|
# as part of this package.
|
|
"""Test the creation of graphics of distribution information.
|
|
|
|
Provides tests for the Graphics.Distribution classes which provide the
|
|
ability to create graphics to show the distribution of information about
|
|
BACs/chromosomes/etc.
|
|
"""
|
|
|
|
# standard library
|
|
import os
|
|
import random
|
|
import unittest
|
|
|
|
from Bio import MissingExternalDependencyError
|
|
|
|
try:
|
|
import reportlab as r
|
|
|
|
del r
|
|
except ImportError:
|
|
raise MissingExternalDependencyError(
|
|
"Install reportlab if you want to use Bio.Graphics."
|
|
) from None
|
|
|
|
# local stuff
|
|
from Bio.Graphics.Distribution import BarChartDistribution
|
|
|
|
# TODO from Bio.Graphics.Distribution import LineDistribution
|
|
from Bio.Graphics.Distribution import DistributionPage
|
|
|
|
|
|
def random_distribution(min=-5.0, max=5.0, total_items=50):
|
|
"""Create a series of random distribution information."""
|
|
num_items = random.randrange(5, total_items)
|
|
all_info = []
|
|
for item in range(num_items):
|
|
new_item = random.uniform(min, max)
|
|
all_info.append(new_item)
|
|
|
|
return all_info
|
|
|
|
|
|
class BarChartTest(unittest.TestCase):
|
|
"""Test display of BarChart distributions on a page."""
|
|
|
|
def setUp(self):
|
|
self.simple_page = os.path.join(os.getcwd(), "Graphics", "simple_bar.pdf")
|
|
self.multi_page = os.path.join(os.getcwd(), "Graphics", "multi_bar.pdf")
|
|
|
|
self.num_multi = 5
|
|
|
|
def test_simple_page(self):
|
|
"""Test displaying a page with single distribution."""
|
|
dist_info = []
|
|
new_info = random_distribution()
|
|
dist_info.append(new_info)
|
|
distribution = BarChartDistribution(dist_info)
|
|
|
|
dist_page = DistributionPage()
|
|
dist_page.distributions.append(distribution)
|
|
|
|
dist_page.draw(self.simple_page, "Test Bar Chart")
|
|
|
|
def test_multi_page(self):
|
|
"""Create a page with multiple distributions on it."""
|
|
dist_page = DistributionPage()
|
|
|
|
dist_page.number_of_columns = 3
|
|
|
|
for multi in range(self.num_multi):
|
|
dist_info = []
|
|
new_info = random_distribution()
|
|
dist_info.append(new_info)
|
|
|
|
distribution = BarChartDistribution(dist_info)
|
|
distribution.chart_title = f"Distribution {multi + 1}"
|
|
dist_page.distributions.append(distribution)
|
|
|
|
dist_page.draw(self.multi_page, "Test Multi Bar Chart")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
runner = unittest.TextTestRunner(verbosity=2)
|
|
unittest.main(testRunner=runner)
|