Add tests for the 0.0 → 0 changes

Author: mhucka

src/openfermion/ops/operators/fermion_operator_test.py

@@ -9,8 +9,11 @@
 #   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 #   See the License for the specific language governing permissions and
 #   limitations under the License.
-"""Tests  fermion_operator.py."""
+
+"""Tests fermion_operator.py."""
+
 import unittest
+import sympy
 
 from openfermion.ops.operators.fermion_operator import FermionOperator
 from openfermion.hamiltonians import number_operator
@@ -78,3 +81,10 @@ def test_is_two_body_number_conserving_three(self):
     def test_is_two_body_number_conserving_out_of_order(self):
         op = FermionOperator(((0, 1), (2, 0), (1, 1), (3, 0)))
         self.assertTrue(op.is_two_body_number_conserving())
+
+    def test_add_sympy_rational(self):
+        """Test adding operators with sympy.Rational coefficients."""
+        a = FermionOperator('0^ 0', sympy.Rational(1, 2))
+        b = FermionOperator('1^ 1', sympy.Rational(1, 2))
+        c = a + b
+        self.assertIsInstance(c.terms[((0, 1), (0, 0))], sympy.Rational)

src/openfermion/ops/operators/symbolic_operator_test.py

@@ -20,6 +20,7 @@
 from openfermion.testing.testing_utils import EqualsTester
 
 from openfermion.ops.operators.symbolic_operator import SymbolicOperator
+from openfermion.ops.operators.fermion_operator import FermionOperator
 
 
 class DummyOperator1(SymbolicOperator):
@@ -687,6 +688,14 @@ def test_add_sympy(self):
         self.assertTrue(a.terms[term_a] - coeff_a == 0)
         self.assertTrue(a.terms[term_b] - coeff_b - 0.5 == 0)
 
+    def test_add_sympy_new_term(self):
+        """Test adding a new term with a sympy coefficient."""
+        x = sympy.Symbol('x')
+        op = FermionOperator('1^', x)
+        op += FermionOperator('2', 2 * x)
+        self.assertEqual(op.terms[((1, 1),)], x)
+        self.assertEqual(op.terms[((2, 0),)], 2 * x)
+
     def test_radd(self):
         term_a = ((1, 1), (3, 0), (8, 1))
         coeff_a = 1