lib/eulim/chemistry/compound.rb in eulim-0.0.10 vs lib/eulim/chemistry/compound.rb in eulim-0.0.11

@@ -1,5 +1,7 @@
+require 'unitwise'
+
 module Eulim
   module Chemistry
     # This class has functionality for compounds
     # Ex: constituent elements, molecular mass, etc
     class Compound
@@ -15,10 +17,10 @@
       end
 
       private
 
       def calculate_molecular_mass
-        @molecular_mass = 0
+        @molecular_mass = Unitwise(0, 'u')
         @constituents.each do |_symbol, info|
           @molecular_mass += info[:element].atomic_mass * info[:atom_count]
         end
         @molecular_mass
       end