def test_draw_pedigree_hardcoded(self):
'''Test drawing a pedigree as an EPS file using hard-coded positions.'''
coord_params = fp.CoordParams()
coord_params.algorithm = 'hardcoded'
coord_params.coord_file = TestFamPlot.__PEDIGREE_POSITIONS
coord_params.balance_marriages = False
out = tempfile.NamedTemporaryFile(suffix='.eps', delete=False)
ped_info = fp.draw_pedigree(TestFamPlot.__PEDIGREE, coord_params,
fp.PlotParams(), fp.DEFAULT_COLORS, out)
expected_coord = TestFamPlot.__read_coord_dict(coord_params.coord_file)
test_util.assert_positions_almost_equal(ped_info.coord_dict, expected_coord,
decimal=10, err_msg='Wrong coordinates')
def test_layout_position(self):
'''Test family computation.'''
positions = pedigree_plot_laplacian._layout_positions(nx.path_graph(10, create_using=nx.DiGraph()))
expected = {0: (-0.44170765403093781 , 9),
1: (-0.39847023129619913 , 8),
2: (-0.31622776601683716 , 7),
3: (-0.20303072371134445 , 6),
4: (-0.069959619570752599, 5),
5: ( 0.069959619570755624, 4),
6: ( 0.20303072371134645 , 3),
7: ( 0.3162277660168385 , 2),
8: ( 0.39847023129620007 , 1),
9: ( 0.44170765403093859 , 0)}
test_util.assert_positions_almost_equal(positions, expected, decimal=10,
err_msg='Wrong positions', allow_flip=True)
def test_draw_pedigree_hardcoded(self):
'''Test drawing a pedigree as an EPS file using hard-coded positions.'''
coord_params = fp.CoordParams()
coord_params.algorithm = 'hardcoded'
coord_params.coord_file = TestFamPlot.__PEDIGREE_POSITIONS
coord_params.balance_marriages = False
out = tempfile.NamedTemporaryFile(suffix='.eps', delete=False)
ped_info = fp.draw_pedigree(TestFamPlot.__PEDIGREE, coord_params,
fp.PlotParams(), fp.DEFAULT_COLORS, out)
expected_coord = TestFamPlot.__read_coord_dict(coord_params.coord_file)
test_util.assert_positions_almost_equal(ped_info.coord_dict,
expected_coord,
decimal=10,
err_msg='Wrong coordinates')
def test_layout_position(self):
'''Test family computation.'''
positions = pedigree_plot_laplacian._layout_positions(
nx.path_graph(10, create_using=nx.DiGraph()))
expected = {
0: (-0.44170765403093781, 9),
1: (-0.39847023129619913, 8),
2: (-0.31622776601683716, 7),
3: (-0.20303072371134445, 6),
4: (-0.069959619570752599, 5),
5: (0.069959619570755624, 4),
6: (0.20303072371134645, 3),
7: (0.3162277660168385, 2),
8: (0.39847023129620007, 1),
9: (0.44170765403093859, 0)
}
test_util.assert_positions_almost_equal(positions,
expected,
decimal=10,
err_msg='Wrong positions',
allow_flip=True)