def test_number_operator_nosite(self):
        op = number_operator(4, parity=-1)
        expected = (FermionOperator(((0, 1), (0, 0))) + FermionOperator(
            ((1, 1), (1, 0))) + FermionOperator(
                ((2, 1), (2, 0))) + FermionOperator(((3, 1), (3, 0))))
        self.assertEqual(op, expected)

        op = number_operator(4, parity=1)
        expected = (BosonOperator(((0, 1), (0, 0))) + BosonOperator(
            ((1, 1), (1, 0))) + BosonOperator(
                ((2, 1), (2, 0))) + BosonOperator(((3, 1), (3, 0))))
        self.assertTrue(op == expected)
 def test_number_operator_nosite(self):
     op = number_operator(4)
     expected = (FermionOperator(((0, 1), (0, 0))) +
                 FermionOperator(((1, 1), (1, 0))) +
                 FermionOperator(((2, 1), (2, 0))) +
                 FermionOperator(((3, 1), (3, 0))))
     self.assertEqual(op, expected)
 def test_number_operator_site(self):
     op = number_operator(3, 2, 1j)
     self.assertTrue(op == FermionOperator(((2, 1), (2, 0))) * 1j)
    def test_fermion_number_operator_site(self):
        op = number_operator(3, 2, 1j, -1)
        self.assertEqual(op, FermionOperator(((2, 1), (2, 0))) * 1j)

        op = number_operator(3, 2, 1j, 1)
        self.assertTrue(op == BosonOperator(((2, 1), (2, 0))) * 1j)