1.1 --- a/test/test_base.py	Tue Feb 08 20:59:56 2011 +0300
     1.2 +++ b/test/test_base.py	Tue Feb 08 21:03:10 2011 +0300
     1.3 @@ -1,23 +1,41 @@
     1.4  import allpy.base as b
     1.5 -import allpy.protein
     1.6 +import allpy.protein as p
     1.7 +from StringIO import StringIO
     1.8  
     1.9  def test_new_monomers():
    1.10  
    1.11 -	try:
    1.12 -		m = b.Monomer.from_code1('A')
    1.13 -	except Exception:
    1.14 -		pass
    1.15 -	else:
    1.16 -		assert False, "base.Monomers must not be constructible from code1"
    1.17 +    try:
    1.18 +        m = b.Monomer.from_code1('A')
    1.19 +    except Exception:
    1.20 +        pass
    1.21 +    else:
    1.22 +        assert False, "base.Monomers must not be constructible from code1"
    1.23  
    1.24 -	m = b.Monomer.from_code3('ALA')
    1.25 -	assert m.__class__.__name__ == "Alanine"
    1.26 -	assert m.code1 == "A"
    1.27 -	assert m.code3 == "ALA"
    1.28 -	assert m.name == "Alanine"
    1.29 +    m = b.Monomer.from_code3('ALA')
    1.30 +    assert m.__class__.__name__ == "Alanine"
    1.31 +    assert m.code1 == "A"
    1.32 +    assert m.code3 == "ALA"
    1.33 +    assert m.name == "Alanine"
    1.34  
    1.35 -	m = b.Monomer.from_name("alaNINE")
    1.36 -	assert m.__class__.__name__ == "Alanine"
    1.37 -	assert m.code1 == "A"
    1.38 -	assert m.code3 == "ALA"
    1.39 -	assert m.name == "Alanine"
    1.40 +    m = b.Monomer.from_name("alaNINE")
    1.41 +    assert m.__class__.__name__ == "Alanine"
    1.42 +    assert m.code1 == "A"
    1.43 +    assert m.code3 == "ALA"
    1.44 +    assert m.name == "Alanine"
    1.45 +
    1.46 +def test_alignment_changes():
    1.47 +
    1.48 +    a = (p.Alignment().
    1.49 +        append_row_from_string("a-------cdef").
    1.50 +        append_row_from_string("ghiklmpq--").
    1.51 +        append_row_from_string("-------------rst"))
    1.52 +
    1.53 +    s = StringIO()
    1.54 +    a.to_file(s)
    1.55 +    assert s.getvalue() == (
    1.56 +        ">\nA-------CDEF----\n"
    1.57 +        ">\nGHIKLMPQ--------\n"
    1.58 +        ">\n-------------RST\n"
    1.59 +    )
    1.60 +
    1.61 +# vim: set et ts=4 sts=4 sw=4: