def test_process_monthly(self): db = config.get_database() dbu = DatabaseUtilities() dbu.delete_all_tables() dbu.create_some_test_well_royalty_masters() dbu.create_some_test_lease_royalty_masters() dbu.create_some_test_leases() dbu.create_some_test_well_lease_link() dbu.create_some_test_monthly() dbu.create_some_test_econdata() dbu.create_some_test_rtp_info() dbu.create_calc() pr = ProcessRoyalties() pr.process_one(4, 201501, "OIL") # Check to see if the oper record exists self.assertEqual(2, db.count("calc")) pr.process_all() self.assertEqual(3, db.count("calc"))