def test_freeze_orbitals_vanishing(self):
op = FermionOperator(((1, 1), (2, 0)))
op_frozen = freeze_orbitals(op, [], [2])
self.assertEquals(len(op_frozen.terms), 0)
def test_freeze_orbitals_nonvanishing(self):
op = FermionOperator(((1, 1), (1, 0), (0, 1), (2, 0)))
op_frozen = freeze_orbitals(op, [1])
expected = FermionOperator(((0, 1), (1, 0)), -1)
self.assertTrue(op_frozen == expected)