Refactor train model exports to make it mockable

Author: maddiedierker

src/oceans/models/index.js

@@ -1,6 +1,6 @@
 import {init as initLoading} from './loading';
 import {init as initWords} from './words';
-import {init as initTraining} from './train';
+import train from './train';
 import {init as initPredicting} from './predict';
 import {init as initPond} from './pond';
 import {Modes} from '../constants';
@@ -17,7 +17,7 @@ export const init = state => {
       initWords();
       break;
     case Modes.Training:
-      initTraining();
+      train.init();
       break;
     case Modes.Predicting:
       initPredicting();

src/oceans/models/train.js

@@ -5,7 +5,7 @@ import SimpleTrainer from '../../utils/SimpleTrainer';
 import SVMTrainer from '../../utils/SVMTrainer';
 import {generateOcean} from '../../utils/generateOcean';
 
-export const init = () => {
+const init = () => {
   const state = getState();
 
   let trainer = state.trainer;
@@ -38,7 +38,7 @@ export const init = () => {
   });
 };
 
-export const onClassifyFish = doesLike => {
+const onClassifyFish = doesLike => {
   const state = getState();
 
   // No-op if animation is currently in progress.
@@ -73,3 +73,8 @@ export const onClassifyFish = doesLike => {
 
   return true;
 };
+
+export default {
+  init,
+  onClassifyFish
+};

src/oceans/ui.jsx

@@ -12,7 +12,7 @@ import {
   resetTraining,
   friendlyNameForFishPart
 } from './helpers';
-import {onClassifyFish} from './models/train';
+import train from './models/train';
 import {arrangeFish} from './models/pond';
 import colors from './colors';
 import aiBotHead from '@public/images/ai-bot/ai-bot-head.png';
@@ -927,7 +927,7 @@ let UnwrappedTrain = class Train extends React.Component {
             style={styles.trainButtonNo}
             onClick={() => {
               this.setState({headOpen: true});
-              return onClassifyFish(false);
+              return train.onClassifyFish(false);
             }}
             sound={'no'}
           >
@@ -939,7 +939,7 @@ let UnwrappedTrain = class Train extends React.Component {
             style={styles.trainButtonYes}
             onClick={() => {
               this.setState({headOpen: true});
-              return onClassifyFish(true);
+              return train.onClassifyFish(true);
             }}
             sound={'yes'}
           >

test/unit/oceans/models/train.test.js

@@ -1,7 +1,7 @@
 const {initFishData} = require('@ml/utils/fishData');
 import {setState, getState, resetState} from '@ml/oceans/state';
 import {ClassType, Modes} from '@ml/oceans/constants';
-import {init, onClassifyFish} from '@ml/oceans/models/train.js';
+import train from '@ml/oceans/models/train.js';
 
 describe('Model quality test', () => {
   beforeAll(() => {
@@ -17,7 +17,7 @@ describe('Model quality test', () => {
   });
 
   test('init state', () => {
-    init();
+    train.init();
     const state = getState();
     expect(state).toBeTruthy();
     expect(state.trainer).toBeTruthy();
@@ -26,11 +26,11 @@ describe('Model quality test', () => {
   });
 
   test('state changes on classify', () => {
-    init();
+    train.init();
     // Set isRunning to false to simulate animation ending
     setState({isRunning: false});
     const previousState = getState();
-    expect(onClassifyFish(true)).toBe(true);
+    expect(train.onClassifyFish(true)).toBe(true);
     const newState = getState();
     expect(newState.trainingIndex).toBe(previousState.trainingIndex + 1);
     expect(newState.yesCount).toBe(previousState.yesCount + 1);