fix: entity tests for version change

Author: Victor Wiebe

src/entities/__tests__/SecurityToken.ts

@@ -1,14 +1,15 @@
 import { BigNumber } from '@polymathnetwork/contract-wrappers';
 import { SecurityToken } from '../SecurityToken';
 import { Context } from '../../Context';
+import { Version } from '~/types';
 
 const params1 = {
   symbol: 'TEST1',
   name: 'Test Token 1',
   address: '0x1',
   owner: '0x3',
   tokenDetails: 'details',
-  version: [new BigNumber(1)],
+  version: Version.V3_1_0,
   granularity: 3,
   totalSupply: new BigNumber(1000),
   currentCheckpoint: 2,

src/entities/__tests__/Shareholders.ts

@@ -6,14 +6,15 @@ import * as createCheckpointProcedure from '../../procedures/CreateCheckpoint';
 import { Shareholders } from '../../entities/SecurityToken/Shareholders';
 import { SubModule } from '../../entities/SecurityToken/SubModule';
 import { SecurityToken } from '../SecurityToken';
+import { Version } from '../../types';
 
 const params1 = {
   symbol: 'TEST1',
   name: 'Test Token 1',
   address: '0x1',
   owner: '0x3',
   tokenDetails: 'details',
-  version: [new BigNumber(1)],
+  version: Version.V3_1_0,
   granularity: 3,
   totalSupply: new BigNumber(1000),
   currentCheckpoint: 2,